游雁
2023-11-20 df03a020f6d8fe4e9b09c1e784fead2852d90bfc
funasr/bin/vad_inference_launch.py
@@ -90,6 +90,7 @@
            preprocess_args=None,
            data_path_and_name_and_type=data_path_and_name_and_type,
            dtype=dtype,
            fs=fs,
            batch_size=batch_size,
            key_file=key_file,
            num_workers=num_workers,
@@ -123,7 +124,7 @@
                vad_results.append(item)
                if writer is not None:
                    ibest_writer["text"][keys[i]] = "{}".format(results[i])
        torch.cuda.empty_cache()
        return vad_results
    return _forward
@@ -189,6 +190,7 @@
            preprocess_args=None,
            data_path_and_name_and_type=data_path_and_name_and_type,
            dtype=dtype,
            fs=fs,
            batch_size=batch_size,
            key_file=key_file,
            num_workers=num_workers,