Pytorch中使用tensorboard时的常见问题分析

---

前言

最近在训练神经网络时，需要可视化看每次训练的accuracy和loss值。一般可以选用matplotlib或tensorboard等。用matplotlib绘制的话会有一些麻烦，还需要调节很多绘图时的参数，因此，我推荐使tensorboard。因为，tensorboard不仅可以可视化训练结果， 而 且还可以输出搭建的网络结构，以及可视化输入数据。本文将简要介绍，作者在使用tensorboard过程中所采的小坑。至于tensorboard的使用步骤，作者会引用几个比较好的博客或视频链接供大家参考！

踩坑1：tensorboard中怎么将训练结果和测试结果同时放置到一张图中做对比？

答：1）解决这个问题这里可以使用writer.add_scalars(self, main_tag, tag_scalar_dict, global_step=None, walltime=None)，可以将你想对比的数据放入一个字典中，作为参数tag_scalar_dict传入add_scalars中。例子如下：

2）另外，可以学习一下writer.add_scalar和add_scalars的区别，有助于理解。参考链接如下：https://blog.csdn.net/sdnuwjw/article/details/107305358

踩坑2：Pytorch中使用tensorboard可视化不显示的问题？

答：1）首先查看，你是否在网络模型中正确使用了Summarywriter，是否保存的有数据文件。
2）一定要让终端路径切换到你所保存的数据文件路径，然后直接输入tensorboard --logdir=./ 这样既可。可以消除，你因路径问题不能正确使用。
注：有些博客说一定要用Googlechrome，才能正常使用。这个说法太绝对，我使用IE浏览器也可以。

擦坑3：Pytorch中使用tensorboard，加载第二次新的数据，却显示的还是第一次的数据结果。

答：作者使用的是pycharm中的终端。在可视化新的数据结果时，需要提前关闭上次打开的所有终端窗口，然后在新的终端窗口中加载即可刷新。

Reference：

这部分我将列出一些类似问题，回答的比较好的链接供大家参考：
[1] 用tensorboard可视化log日志: https://blog.csdn.net/qq_45488242/article/details/122787702
[2] （视频看后十分钟即可）：https://www.bilibili.com/video/BV1Qf4y1C7kz/?spm_id_from=333.880.my_history.page.click&vd_source=78b6dcb9302de61d9617ef8e8d444475
[3] (这是英语中文字幕视频，介绍的较为详细)：https://www.bilibili.com/video/BV19Y4y187L6/?spm_id_from=333.880.my_history.page.click&vd_source=78b6dcb9302de61d9617ef8e8d444475

总结

主要介绍一些pytorch中使用tensorboard所遇到的一些问题，通过查阅资料和实践，表达了自己的解决方案和思想。如有错误的地方，感谢您的留言纠正！
除此之外，作者还总结了部分机器学习面试相关的重要基础知识点总结，可供大家参考哦，链接如下：机器/深度学习重要基础面试知识点总结