嘉渊
2023-04-25 ee9e86d5f0b5f546f77de1ea8b59cdb67aaad478
update
2个文件已修改
3 ■■■■ 已修改文件
funasr/bin/train.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/datasets/small_datasets/preprocessor.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/train.py
@@ -527,6 +527,7 @@
                                                                   distributed_option.dist_rank,
                                                                   distributed_option.local_rank))
    logging.info(pytorch_cudnn_version())
    logging.info("Args: {}".format(args))
    logging.info(model_summary(model))
    logging.info("Optimizer: {}".format(optimizers))
    logging.info("Scheduler: {}".format(schedulers))
funasr/datasets/small_datasets/preprocessor.py
@@ -820,7 +820,7 @@
def build_preprocess(args, train):
    if args.use_preprocessor:
    if not args.use_preprocessor:
        return None
    if args.task_name in ["asr", "data2vec", "diar", "sv"]:
        retval = CommonPreprocessor(