mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-04 14:44:18 +00:00
Minor fix
This commit is contained in:
parent
36726ca188
commit
b7ecb3b620
@ -650,7 +650,7 @@ def train_one_epoch(
|
||||
optimizer.step()
|
||||
|
||||
if old_parameters is not None:
|
||||
deltas = optim_step_and_measure_param_change(old_parameters)
|
||||
deltas = optim_step_and_measure_param_change(model, old_parameters)
|
||||
tb_writer.add_scalars(
|
||||
"train/relative_param_change_per_minibatch",
|
||||
deltas,
|
||||
|
Loading…
x
Reference in New Issue
Block a user