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 batching code for scalars

Browse Source
This commit is contained in:
Daniel Povey 2022-07-12 08:36:45 +08:00
parent 25cb8308d5
commit 8c44ff26f7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
View File

@ -85,7 +85,7 @@ class BatchedOptimizer(Optimizer):
yield p_stacked, state # <-- calling code will do the actual optimization here! yield p_stacked, state # <-- calling code will do the actual optimization here!
# Now un-stack the parameter changes # Now un-stack the parameter changes
for i,p in enumerate(batch): for i,p in enumerate(batch):
p[:] = p_stacked[i] p.copy_(p_stacked[i])