gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-03-12 21:02:50 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fix epoch updates count logic as drop_last

Browse Source
This commit is contained in:
Yushen CHEN
2026-01-21 21:36:57 +08:00
committed by GitHub
parent bc1df7a4fa
commit 37a2633d35
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/f5_tts/scripts/count_max_epoch_precise.py
View File

@@ -20,13 +20,14 @@ batch_sampler = DynamicBatchSampler(
random_seed=666,
drop_residual=False,
)
updates_per_epoch = int(len(batch_sampler) / gpus)
print(
f"One epoch has {len(batch_sampler) / gpus} updates if gpus={gpus}, with "
f"One epoch has {updates_per_epoch} updates if gpus={gpus}, with "
f"batch_size_per_gpu={batch_size_per_gpu} (frames) & "
f"max_samples_per_gpu={max_samples_per_gpu}."
)
print(
f"If gpus={gpus}, for max_updates={max_updates} "
f"should set epoch={math.ceil(max_updates / len(batch_sampler) * gpus)}."
f"should set epoch={math.ceil(max_updates / updates_per_epoch)}."
)