首页 PaddleCV 帖子详情
visualdl画的loss曲线横坐标怎么改为epoch?
收藏
快速回复
PaddleCV 问答语义分割目标识别学习资料 362 1
visualdl画的loss曲线横坐标怎么改为epoch?
收藏
快速回复
PaddleCV 问答语义分割目标识别学习资料 362 1

请问大家用visualdl可视化训练过程的数据时,怎么把loss函数的横坐标改为epoch呢?只能自己画吗?

0
收藏
回复
全部评论(1)
时间顺序
宇宙物语
#2 回复于2023-09

# 从 visualdl 库中引入 LogWriter 类
from visualdl import LogWriter
# 创建 LogWriter 对象,指定 logdir 参数,如果指定路径不存在将会创建一个文件夹
logwriter = LogWriter(logdir='./runs/mnist_experiment')

#记录当前训练 Loss 到 VisualDL
logwriter.add_scalar("train_avg_loss", value=avg_loss.numpy(), step=batch_id+epoch_id*(train_batchs_per_epoch))

 

在每个epoch后调用即可

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户