mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in get_adjusted_batch_count

Browse Source
This commit is contained in:
Daniel Povey 2023-01-07 17:45:22 +08:00
parent b3527fe4ac
commit c7107ead64
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless7/train.py
View File

@ -100,8 +100,9 @@ def get_adjusted_batch_count(
params: AttributeDict) -> float:
# returns the number of batches we would have used so far if we had used the reference
# duration. This is for purposes of set_batch_count().
return (params.batch_idx_train * params.ref_duration /
(params.max_duration * params.world_size))
return (params.batch_idx_train * (params.max_duration * params.world_size) /
params.ref_duration)