gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-09 03:43:19 -08:00
Code Issues Packages Projects Releases Wiki Activity

Update trainer.py

Browse Source
This commit is contained in:
Zhikang Niu
2024-10-13 10:04:13 +08:00
committed by GitHub
parent 77abf4e98a
commit edc189fa96
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model/trainer.py
View File

@@ -172,7 +172,7 @@ class Trainer:
train_dataloader = DataLoader(train_dataset, collate_fn=collate_fn, num_workers=num_workers, pin_memory=True,
batch_sampler=batch_sampler)
else:
raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but recieved {self.batch_size_type}")
raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but received {self.batch_size_type}")
# accelerator.prepare() dispatches batches to devices;
# which means the length of dataloader calculated before, should consider the number of devices