visualdl画的loss曲线横坐标怎么改为epoch?
收藏
请问大家用visualdl可视化训练过程的数据时,怎么把loss函数的横坐标改为epoch呢?只能自己画吗?
0
收藏
请登录后评论
# 从 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后调用即可