mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 09:32:20 +00:00
minor fix, remove numel = p.numel() in optim.py (#967)
This commit is contained in:
parent
f260a09ed4
commit
7155769c19
@ -282,7 +282,6 @@ class ScaledAdam(BatchedOptimizer):
|
||||
|
||||
batch_size = p.shape[0]
|
||||
numel = p.numel() // batch_size
|
||||
numel = p.numel()
|
||||
|
||||
if numel > 1:
|
||||
# "param_rms" just periodically records the scalar root-mean-square value of
|
||||
|
Loading…
x
Reference in New Issue
Block a user