FUNASR训练
parent: a6889a31 | 补丁 | 提交 | ignore whitespace
shixian.shi
2023-05-04 bc57005507aad83ed7c092e3a3ea8e994451d105 
update
1个文件已修改
3 ■■■■■ 已修改文件
funasr/bin/build_trainer.py 3 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/bin/build_trainer.py


@@ -132,7 +132,8 @@


        if args.dataset_type == "small":


            args.batch_bins = batch_bins


        elif args.dataset_type == "large":


            args.dataset_conf["batch_conf"]["batch_size"] = batch_bins


            # args.dataset_conf["batch_conf"]["batch_size"] = batch_bins


            pass


        else:


            raise ValueError(f"Not supported dataset_type={args.dataset_type}")


    if args.normalize in ["null", "none", "None"]: