fix unit test

pull/1488/head
huangyuxin 3 years ago
parent 48e610755d
commit cea5728dd7

@ -191,7 +191,8 @@ class TestDeepSpeech2StaticModelOnline(unittest.TestCase):
def setUp(self): def setUp(self):
export_prefix = "exp/deepspeech2_online/checkpoints/test_export" export_prefix = "exp/deepspeech2_online/checkpoints/test_export"
os.makedirs( os.path.dirname(export_prefix), mode=0o755) if not os.path.exists(os.path.dirname(export_prefix)):
os.makedirs(os.path.dirname(export_prefix), mode=0o755)
infer_model = DeepSpeech2InferModelOnline( infer_model = DeepSpeech2InferModelOnline(
feat_size=161, feat_size=161,
dict_size=4233, dict_size=4233,

@ -0,0 +1,3 @@
mkdir -p test_data
wget -P test_data https://paddlespeech.bj.bcebos.com/datasets/unit_test/asr/static_ds2online_inputs.pickle
python deepspeech2_online_model_test.py
Loading…
Cancel
Save