require minimum of 1 second durations for training because of my slop code auto-transposing that I don't wanna fix right now

This commit is contained in:
mrq 2025-02-26 22:00:33 -06:00
parent cbd4d7d7f4
commit 06ef3daf3c
2 changed files with 4 additions and 1 deletions

View File

@ -707,6 +707,8 @@ def _get_duration_map( type="training" ):
return _durations_map[type] if type in _durations_map else {}
def _load_paths(dataset, type="training", silent=not is_global_leader(), dataset_hash_key=None):
assert cfg.dataset.min_duration >= 1.0, "Minimum duration too low."
if not dataset_hash_key:
dataset_hash_key = cfg.dataset.hash_key(sorted(dataset))
@ -862,6 +864,7 @@ class Dataset(_Dataset):
extra_paths_by_spkr_name: dict[str, list] = {},
):
super().__init__()
self._head = None
self.sampler = None

View File

@ -35,7 +35,7 @@ def train_feeder(engine, batch, teacher=None):
engine.current_batch_size = batch_size
output = engine(
phn_list=batch["phns"],
phns_list=batch["phns"],
proms_list=batch["proms"],
resps_list=batch["resps"],
lang_list=batch["lang"],