funasr/bin/punc_train.py
@@ -44,4 +44,10 @@ else: args.distributed = False if args.dataset_type == "small": if args.batch_size is not None: args.batch_size = args.batch_size * args.ngpu * args.num_worker_count if args.batch_bins is not None: args.batch_bins = args.batch_bins * args.ngpu * args.num_worker_count main(args=args)