Minor fix

This commit is contained in:
pkufool 2022-04-07 13:03:10 +08:00
parent 36726ca188
commit b7ecb3b620

View File

@ -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,