This commit is contained in:
shixian.shi 2023-05-04 19:36:23 +08:00
parent a6889a3170
commit bc57005507

View File

@ -132,7 +132,8 @@ def build_trainer(modelscope_dict,
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"]: