| | |
|---|
| | | device = kwargs.get("device", "cuda") |
|---|
| | | if not torch.cuda.is_available() or kwargs.get("ngpu", 0): |
|---|
| | | device = "cpu" |
|---|
| | | # kwargs["batch_size"] = 1 |
|---|
| | | kwargs["batch_size"] = 1 |
|---|
| | | kwargs["device"] = device |
|---|
| | | |
|---|
| | | if kwargs.get("ncpu", None): |
|---|
| | |
|---|
| | | time_speech_total_per_sample = speech_lengths/16000 |
|---|
| | | time_speech_total_all_samples += time_speech_total_per_sample |
|---|
| | | |
|---|
| | | all_segments = [] |
|---|
| | | for j, _ in enumerate(range(0, n)): |
|---|
| | | batch_size_ms_cum += (sorted_data[j][0][1] - sorted_data[j][0][0]) |
|---|
| | | if j < n - 1 and ( |
|---|
| | |
|---|
| | | speech_j, speech_lengths_j = slice_padding_audio_samples(speech, speech_lengths, sorted_data[beg_idx:end_idx]) |
|---|
| | | results = self.inference(speech_j, input_len=None, model=model, kwargs=kwargs, **cfg) |
|---|
| | | if self.spk_model is not None: |
|---|
| | | all_segments = [] |
|---|
| | | |
|---|
| | | # compose vad segments: [[start_time_sec, end_time_sec, speech], [...]] |
|---|
| | | for _b in range(len(speech_j)): |
|---|
| | | vad_segments = [[sorted_data[beg_idx:end_idx][_b][0][0]/1000.0, \ |
|---|
