Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 286
Star 8
Fork 1
P
PaddleSeg
提交 39e5ae8a 编写于 作者: C chenguowei01
浏览文件
操作

update train.py

上级 45983e33
显示空白变更内容
内联 并排
Showing with 2 addition and 1 deletion
dygraph/core/train.py
浏览文件 @ 39e5ae8a
...@@ -117,7 +117,8 @@ def train(model, ...@@ -117,7 +117,8 @@ def train(model,
avg_loss * nranks, lr, avg_train_batch_cost, avg_loss * nranks, lr, avg_train_batch_cost,
avg_train_reader_cost, eta)) avg_train_reader_cost, eta))
if use_vdl: 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/lr', lr, num_steps)
log_writer.add_scalar('Train/batch_cost', log_writer.add_scalar('Train/batch_cost',
avg_train_batch_cost, num_steps) avg_train_batch_cost, num_steps)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或