funasr/datasets/small_datasets/sequence_iter_factory.py
@@ -66,8 +66,9 @@ batch_bins=dataset_conf["batch_conf"]["batch_size"] * args.ngpu, shape_files=shape_files, sort_in_batch=dataset_conf["sort_in_batch"] if hasattr(dataset_conf, "sort_in_batch") else "descending", sort_batch=dataset_conf["sort_batch"] if hasattr(dataset_conf, "sort_batch") else "ascending", sort_batch=dataset_conf["sort_batch"] if hasattr(dataset_conf, "sort_batch") else "descending", drop_last=False, min_batch_size=torch.distributed.get_world_size(), padding=True, )
