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()
|
optimizer.step()
|
||||||
|
|
||||||
if old_parameters is not None:
|
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(
|
tb_writer.add_scalars(
|
||||||
"train/relative_param_change_per_minibatch",
|
"train/relative_param_change_per_minibatch",
|
||||||
deltas,
|
deltas,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user