python/FunASR-XL.git

			@@ -445,6 +445,12 @@
			help='Perform on "collect stats" mode',
			)
			group.add_argument(
			"--mc",
			type=bool,
			default=False,
			help="MultiChannel input",
			)
			group.add_argument(
			"--write_collected_feats",
			type=str2bool,
			default=False,
			@@ -635,8 +641,8 @@
			group.add_argument(
			"--init_param",
			type=str,
			action="append",
			default=[],
			nargs="*",
			help="Specify the file path used for initialization of parameters. "
			"The format is '<file_path>:<src_key>:<dst_key>:<exclude_keys>', "
			"where file_path is the model file path, "
			@@ -662,7 +668,7 @@
			"--freeze_param",
			type=str,
			default=[],
			nargs="*",
			action="append",
			help="Freeze parameters",
			)

			@@ -1153,10 +1159,10 @@
			elif args.distributed and args.simple_ddp:
			distributed_option.init_torch_distributed_pai(args)
			args.ngpu = dist.get_world_size()
			if args.dataset_type == "small":
			if args.dataset_type == "small" and args.ngpu > 0:
			if args.batch_size is not None:
			args.batch_size = args.batch_size * args.ngpu
			if args.batch_bins is not None:
			if args.batch_bins is not None and args.ngpu > 0:
			args.batch_bins = args.batch_bins * args.ngpu

			# filter samples if wav.scp and text are mismatch
			@@ -1316,6 +1322,7 @@
			data_path_and_name_and_type=args.train_data_path_and_name_and_type,
			key_file=train_key_file,
			batch_size=args.batch_size,
			mc=args.mc,
			dtype=args.train_dtype,
			num_workers=args.num_workers,
			allow_variable_data_keys=args.allow_variable_data_keys,
			@@ -1327,6 +1334,7 @@
			data_path_and_name_and_type=args.valid_data_path_and_name_and_type,
			key_file=valid_key_file,
			batch_size=args.valid_batch_size,
			mc=args.mc,
			dtype=args.train_dtype,
			num_workers=args.num_workers,
			allow_variable_data_keys=args.allow_variable_data_keys,