文章目录
- 1准备工作
-
- 1.1 可视化工具-Tensorboard调用
- 1.2 查看训练文件
- 2 Tensorboard在线查看训练曲线
-
- 2.1 查看训练曲线 平滑度调节 SCALARS 训练曲线
- 3 将多条曲线叠加在一张图上显示
- 4 多次训练曲线 放在一张图上对比
1准备工作
1.1 可视化工具-Tensorboard调用
Tensorboard模块安装与调用参考:PyTorch可视化工具-Tensorboard在线查看训练曲线、网络结构图等
本代码已经导入了,Tensorboard,并且在相应的地方调用了,详情可看本文附带的代码,有注释。(不懂的可参考上面链接)
配置好代码之后,正常的训练就行了
训练教程