add speed to vdl

9148fa5c · chenguowei01 · dabacffe · 9148fa5c
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

dygraph/train.py dygraph/train.py +2 -0

未找到文件。
--- 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()