改为每个epoch显示后,test的纵坐标的数值显示有问题

来源:6-8 PyTorch搭建cifar10训练脚本-tensorboard记录LOG(下)

慕用4598413

2023-07-12


根据老师的改动把test_correct和test_loss改为每个epoch显示后,为什么这里的纵坐标的数值变成6,9,10这样了?

写回答

1回答

慕用4598413

提问者

2023-07-12

问题已解决,注意!视频中14分56秒的两行writer.add_scalar未修改完全,视频中未显示正确代码, 这里应为:

writer.add_scalar("test loss", test_loss, global_step=epoch + 1)
writer.add_scalar("test correct", test_correct, global_step=epoch + 1)


0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程