diff --git a/dygraph/train.py b/dygraph/train.py index 5504816541e46fd84bf118e6f62d9feea0e5a086..a95fd090a77eca28dc5cf3bf745a7451dd8ecc08 100644 --- a/dygraph/train.py +++ b/dygraph/train.py @@ -212,6 +212,8 @@ def train(model, if use_vdl: log_writer.add_scalar('Train/loss', avg_loss, num_steps) log_writer.add_scalar('Train/lr', lr, num_steps) + log_writer.add_scalar('Train/time_step', time_step, + num_steps) avg_loss = 0.0 timer.restart()