Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10509580master
| @@ -9,9 +9,8 @@ from modelscope.utils.test_utils import test_level | |||||
| TEST_SPEECH_FILE = 'data/test/audios/3ch_nihaomiya.wav' | TEST_SPEECH_FILE = 'data/test/audios/3ch_nihaomiya.wav' | ||||
| TEST_SPEECH_FILE_MONO = 'data/test/audios/1ch_nihaomiya.wav' | TEST_SPEECH_FILE_MONO = 'data/test/audios/1ch_nihaomiya.wav' | ||||
| TEST_SPEECH_URL = 'https://modelscope.cn/api/v1/models/damo/' \ | |||||
| 'speech_dfsmn_kws_char_farfield_16k_nihaomiya/repo' \ | |||||
| '?Revision=master&FilePath=examples/3ch_nihaomiya.wav' | |||||
| TEST_SPEECH_URL = 'https://modelscope.oss-cn-beijing.aliyuncs.com/' \ | |||||
| 'test/audios/3ch_nihaomiya.wav' | |||||
| class KWSFarfieldTest(unittest.TestCase): | class KWSFarfieldTest(unittest.TestCase): | ||||
| @@ -11,17 +11,14 @@ from modelscope.utils.test_utils import test_level | |||||
| NEAREND_MIC_FILE = 'data/test/audios/nearend_mic.wav' | NEAREND_MIC_FILE = 'data/test/audios/nearend_mic.wav' | ||||
| FAREND_SPEECH_FILE = 'data/test/audios/farend_speech.wav' | FAREND_SPEECH_FILE = 'data/test/audios/farend_speech.wav' | ||||
| NEAREND_MIC_URL = 'https://modelscope.cn/api/v1/models/damo/' \ | |||||
| 'speech_dfsmn_aec_psm_16k/repo?Revision=master' \ | |||||
| '&FilePath=examples/nearend_mic.wav' | |||||
| FAREND_SPEECH_URL = 'https://modelscope.cn/api/v1/models/damo/' \ | |||||
| 'speech_dfsmn_aec_psm_16k/repo?Revision=master' \ | |||||
| '&FilePath=examples/farend_speech.wav' | |||||
| NEAREND_MIC_URL = 'https://modelscope.oss-cn-beijing.aliyuncs.com/' \ | |||||
| 'test/audios/nearend_mic.wav' | |||||
| FAREND_SPEECH_URL = 'https://modelscope.oss-cn-beijing.aliyuncs.com/' \ | |||||
| 'test/audios/farend_speech.wav' | |||||
| NOISE_SPEECH_FILE = 'data/test/audios/speech_with_noise.wav' | NOISE_SPEECH_FILE = 'data/test/audios/speech_with_noise.wav' | ||||
| NOISE_SPEECH_URL = 'https://modelscope.cn/api/v1/models/damo/' \ | |||||
| 'speech_frcrn_ans_cirm_16k/repo?Revision=master' \ | |||||
| '&FilePath=examples/speech_with_noise.wav' | |||||
| NOISE_SPEECH_URL = 'https://modelscope.oss-cn-beijing.aliyuncs.com/' \ | |||||
| 'test/audios/speech_with_noise.wav' | |||||
| class SpeechSignalProcessTest(unittest.TestCase, DemoCompatibilityCheck): | class SpeechSignalProcessTest(unittest.TestCase, DemoCompatibilityCheck): | ||||
| @@ -17,7 +17,6 @@ SEGMENT_LENGTH_TEST = 640 | |||||
| class TestANSTrainer(unittest.TestCase): | class TestANSTrainer(unittest.TestCase): | ||||
| REVISION = 'beta' | |||||
| def setUp(self): | def setUp(self): | ||||
| self.tmp_dir = tempfile.TemporaryDirectory().name | self.tmp_dir = tempfile.TemporaryDirectory().name | ||||
| @@ -25,7 +24,7 @@ class TestANSTrainer(unittest.TestCase): | |||||
| os.makedirs(self.tmp_dir) | os.makedirs(self.tmp_dir) | ||||
| self.model_id = 'damo/speech_frcrn_ans_cirm_16k' | self.model_id = 'damo/speech_frcrn_ans_cirm_16k' | ||||
| cfg = read_config(self.model_id, revision=self.REVISION) | |||||
| cfg = read_config(self.model_id) | |||||
| cfg.train.max_epochs = 2 | cfg.train.max_epochs = 2 | ||||
| cfg.train.dataloader.batch_size_per_gpu = 1 | cfg.train.dataloader.batch_size_per_gpu = 1 | ||||
| self.cfg_file = os.path.join(self.tmp_dir, 'train_config.json') | self.cfg_file = os.path.join(self.tmp_dir, 'train_config.json') | ||||
| @@ -48,7 +47,6 @@ class TestANSTrainer(unittest.TestCase): | |||||
| def test_trainer(self): | def test_trainer(self): | ||||
| kwargs = dict( | kwargs = dict( | ||||
| model=self.model_id, | model=self.model_id, | ||||
| model_revision=self.REVISION, | |||||
| train_dataset=self.dataset, | train_dataset=self.dataset, | ||||
| eval_dataset=self.dataset, | eval_dataset=self.dataset, | ||||
| max_epochs=2, | max_epochs=2, | ||||
| @@ -16,7 +16,6 @@ NOISE_2CH_FILE = 'data/test/audios/noise_2ch.wav' | |||||
| class TestKwsFarfieldTrainer(unittest.TestCase): | class TestKwsFarfieldTrainer(unittest.TestCase): | ||||
| REVISION = 'beta' | |||||
| def setUp(self): | def setUp(self): | ||||
| self.tmp_dir = tempfile.TemporaryDirectory().name | self.tmp_dir = tempfile.TemporaryDirectory().name | ||||
| @@ -70,7 +69,6 @@ class TestKwsFarfieldTrainer(unittest.TestCase): | |||||
| kwargs = dict( | kwargs = dict( | ||||
| model=self.model_id, | model=self.model_id, | ||||
| work_dir=self.tmp_dir, | work_dir=self.tmp_dir, | ||||
| model_revision=self.REVISION, | |||||
| workers=2, | workers=2, | ||||
| max_epochs=2, | max_epochs=2, | ||||
| train_iters_per_epoch=2, | train_iters_per_epoch=2, | ||||