update

嘉渊

2023-04-25 439e2deddea8b4beb85346feb7289f7f37f188d9

 funasr/bin/train.py

@@ -242,6 +242,12 @@
        help="Enable resuming if checkpoint is existing",
    )
    parser.add_argument(
        "--train_dtype",
        default="float32",
        choices=["float16", "float32", "float64"],
        help="Data type for training.",
    )
    parser.add_argument(
        "--use_amp",
        type=str2bool,
        default=False,
@@ -298,6 +304,12 @@
        help="whether to use dataloader for large dataset",
    )
    parser.add_argument(
        "--dataset_conf",
        action=NestedDictAction,
        default=dict(),
        help=f"The keyword arguments for dataset",
    )
    parser.add_argument(
        "--train_data_file",
        type=str,
        default=None,