| | |
| | | |
| | | def test_fbank_cpu_infer(): |
| | | diar_config_path = "config_fbank.yaml" |
| | | diar_model_path = "sond.pth" |
| | | diar_model_path = "sond.pb" |
| | | output_dir = "./outputs" |
| | | data_path_and_name_and_type = [ |
| | | ("data/unit_test/test_feats.scp", "speech", "kaldi_ark"), |
| | |
| | | |
| | | def test_fbank_gpu_infer(): |
| | | diar_config_path = "config_fbank.yaml" |
| | | diar_model_path = "sond.pth" |
| | | diar_model_path = "sond.pb" |
| | | output_dir = "./outputs" |
| | | data_path_and_name_and_type = [ |
| | | ("data/unit_test/test_feats.scp", "speech", "kaldi_ark"), |
| | |
| | | |
| | | def test_wav_gpu_infer(): |
| | | diar_config_path = "config.yaml" |
| | | diar_model_path = "sond.pth" |
| | | diar_model_path = "sond.pb" |
| | | output_dir = "./outputs" |
| | | data_path_and_name_and_type = [ |
| | | ("data/unit_test/test_wav.scp", "speech", "sound"), |
| | |
| | | |
| | | def test_without_profile_gpu_infer(): |
| | | diar_config_path = "config.yaml" |
| | | diar_model_path = "sond.pth" |
| | | diar_model_path = "sond.pb" |
| | | output_dir = "./outputs" |
| | | raw_inputs = [[ |
| | | "data/unit_test/raw_inputs/record.wav", |