From b0c156a5373b25cef24db787b201cd502030a34f Mon Sep 17 00:00:00 2001 From: babysor00 Date: Sun, 17 Oct 2021 17:21:49 +0800 Subject: [PATCH] Add new dataset support to preprocess parameter --- pre.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pre.py b/pre.py index 2ea21ed..17fd0f7 100644 --- a/pre.py +++ b/pre.py @@ -41,7 +41,7 @@ if __name__ == "__main__": "Use this option when dataset does not include alignments\ (these are used to split long audio files into sub-utterances.)") parser.add_argument("-d", "--dataset", type=str, default="aidatatang_200zh", help=\ - "Name of the dataset to process, allowing values: magicdata, aidatatang_200zh, aishell3.") + "Name of the dataset to process, allowing values: magicdata, aidatatang_200zh, aishell3, data_aishell.") parser.add_argument("-e", "--encoder_model_fpath", type=Path, default="encoder/saved_models/pretrained.pt", help=\ "Path your trained encoder model.") parser.add_argument("-ne", "--n_processes_embed", type=int, default=1, help=\