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=\