update train.py

39e5ae8a · chenguowei01 · 45983e33 · 39e5ae8a
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

dygraph/core/train.py dygraph/core/train.py +2 -1

未找到文件。
--- a/dygraph/core/train.py
+++ b/dygraph/core/train.py
@@ -117,7 +117,8 @@ def train(model,
                            avg_loss * nranks, lr, avg_train_batch_cost,
                            avg_train_reader_cost, eta))
                if use_vdl:
-                    log_writer.add_scalar('Train/loss', avg_loss, num_steps)
+                    log_writer.add_scalar('Train/loss', avg_loss * nranks,
+                                          num_steps)
                    log_writer.add_scalar('Train/lr', lr, num_steps)
                    log_writer.add_scalar('Train/batch_cost',
                                          avg_train_batch_cost, num_steps)