改为每个epoch显示后,test的纵坐标的数值显示有问题
来源:6-8 PyTorch搭建cifar10训练脚本-tensorboard记录LOG(下)

慕用4598413
2023-07-12
写回答
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)
00
相似问题