update

嘉渊

2023-08-01 7ea3836893bfdf1aac03952bb1ff2da2c6ef6e57

 funasr/datasets/small_datasets/sequence_iter_factory.py

@@ -57,7 +57,7 @@
            data_path_and_name_and_type,
            preprocess=preprocess_fn,
            dest_sample_rate=dest_sample_rate,
            speed_perturb=args.speed_perturb,
            speed_perturb=args.speed_perturb if mode=="train" else None,
        )

        # sampler
@@ -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,
        )