From d9ad40bf6f5d344b2b07ba1a1b5a819bd4bb836d Mon Sep 17 00:00:00 2001 From: speech_asr Date: Mon, 17 Apr 2023 11:45:41 +0800 Subject: [PATCH] update --- funasr/utils/prepare_data.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/funasr/utils/prepare_data.py b/funasr/utils/prepare_data.py index a8c96ebc1..a0d97f655 100644 --- a/funasr/utils/prepare_data.py +++ b/funasr/utils/prepare_data.py @@ -63,13 +63,10 @@ def calc_shape(args, dataset, nj=32): if os.path.exists(shape_path): print('Shape file for small dataset already exists.') return - os.makedirs(shape_path, exist_ok=True) + split_shape_path = os.path.join(args.data_dir, dataset, "shape_files") - if os.path.exists(shape_path): - assert os.path.exists(os.path.join(args.data_dir, dataset, "speech_shape")) - print('Shape file for small dataset already exists.') - return - os.makedirs(shape_path, exist_ok=True) + if os.path + os.makedirs(split_shape_path, exist_ok=True) # split wav_scp_file = os.path.join(args.data_dir, dataset, "wav.scp")