嘉渊
2023-04-25 c33dc9585c2b1b3b78a10e9a7f69d7cb2fa7f7af
update
1个文件已修改
6 ■■■■■ 已修改文件
funasr/bin/train.py 6 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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,