diff options
-rw-r--r-- | swr2_asr/train.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/swr2_asr/train.py b/swr2_asr/train.py index 3ed3ac8..ffdae73 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -187,7 +187,7 @@ def main(config_path: str): dataset_config = config_dict.get("dataset", {}) tokenizer_config = config_dict.get("tokenizer", {}) checkpoints_config = config_dict.get("checkpoints", {}) - + if not os.path.isdir(dataset_config["dataset_root_path"]): os.makedirs(dataset_config["dataset_root_path"]) |