mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +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]
|
batch_size = p.shape[0]
|
||||||
numel = p.numel() // batch_size
|
numel = p.numel() // batch_size
|
||||||
numel = p.numel()
|
|
||||||
|
|
||||||
if numel > 1:
|
if numel > 1:
|
||||||
# "param_rms" just periodically records the scalar root-mean-square value of
|
# "param_rms" just periodically records the scalar root-mean-square value of
|
||||||
|
Loading…
x
Reference in New Issue
Block a user