shixian.shi
2023-05-04 6482fb94a5f84f98967d3274792772b0a0ab87a2
update
1个文件已修改
5 ■■■■■ 已修改文件
funasr/bin/build_trainer.py 5 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/build_trainer.py
@@ -131,9 +131,8 @@
    if batch_bins is not None:
        if args.dataset_type == "small":
            args.batch_bins = batch_bins
        elif args.dataset_type == "large":
            # args.dataset_conf["batch_conf"]["batch_size"] = batch_bins
            pass
        elif args.dataset_type == "large" and "batch_size" not in args.dataset_conf["batch_conf"]:
            args.dataset_conf["batch_conf"]["batch_size"] = batch_bins
        else:
            raise ValueError(f"Not supported dataset_type={args.dataset_type}")
    if args.normalize in ["null", "none", "None"]: