游雁
2023-05-16 4cd79db451786548d8a100f25c3b03da0eb30f4b
funasr/bin/build_trainer.py
@@ -1,3 +1,8 @@
# -*- encoding: utf-8 -*-
#!/usr/bin/env python3
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
#  MIT License  (https://opensource.org/licenses/MIT)
import os
import yaml
@@ -132,8 +137,7 @@
        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
            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"]: