update

嘉渊

2023-07-27 eedc4231b0f4ec1c6ab22fac163b1512d7e7f4ce

update

 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,
        )