Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]test: use 'master' branch in training test

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10509580
This commit is contained in:
bin.xue
2022-10-25 12:10:07 +08:00
committed by yingda.chen
parent a1738690c9
commit 525fa3ea89
4 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/pipelines/test_key_word_spotting_farfield.py
View File

@@ -9,9 +9,8 @@ from modelscope.utils.test_utils import test_level
TEST_SPEECH_FILE = 'data/test/audios/3ch_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):

15
tests/pipelines/test_speech_signal_process.py
View File

@@ -11,17 +11,14 @@ from modelscope.utils.test_utils import test_level
NEAREND_MIC_FILE = 'data/test/audios/nearend_mic.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_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):

4
tests/trainers/audio/test_ans_trainer.py
View File

@@ -17,7 +17,6 @@ SEGMENT_LENGTH_TEST = 640
class TestANSTrainer(unittest.TestCase):
REVISION = 'beta'
def setUp(self):
self.tmp_dir = tempfile.TemporaryDirectory().name
@@ -25,7 +24,7 @@ class TestANSTrainer(unittest.TestCase):
os.makedirs(self.tmp_dir)
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.dataloader.batch_size_per_gpu = 1
self.cfg_file = os.path.join(self.tmp_dir, 'train_config.json')
@@ -48,7 +47,6 @@ class TestANSTrainer(unittest.TestCase):
def test_trainer(self):
kwargs = dict(
model=self.model_id,
model_revision=self.REVISION,
train_dataset=self.dataset,
eval_dataset=self.dataset,
max_epochs=2,

2
tests/trainers/audio/test_kws_farfield_trainer.py
View File

@@ -16,7 +16,6 @@ NOISE_2CH_FILE = 'data/test/audios/noise_2ch.wav'
class TestKwsFarfieldTrainer(unittest.TestCase):
REVISION = 'beta'
def setUp(self):
self.tmp_dir = tempfile.TemporaryDirectory().name
@@ -70,7 +69,6 @@ class TestKwsFarfieldTrainer(unittest.TestCase):
kwargs = dict(
model=self.model_id,
work_dir=self.tmp_dir,
model_revision=self.REVISION,
workers=2,
max_epochs=2,
train_iters_per_epoch=2,