PyTorch + visdom + netron 实现网络数据和模型结构的可视化
训练网络时,想查看网络模型结构,网上有各种方法。
有些方法复杂,但只需要一种库就可以;有些方法虽然简单,但需要多种库配合使用。
总结了一下,列出了个人觉得 安装最简单,使用和配置也很简洁的方法。
推荐使用visdom(数据可视化) + netron(模型结构可视化)
一、数据可视化-visdom
参考:PyTorch 训练可视化教程 visdom
在训练模型时,查看损失函数等数据的可视化可以使用matplotlib
画图,也可以使用visdom
。
visdom
安装很简单:
pip install visdom
使用前需要先开启监听命令(在cmd
命令行下):
python -m visdom.server # 或者直接visdom
然后浏览器打开如下地址,就可以查看了:
http://localhost:8097