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, | ||||