PyTorch下的可视化工具（网络结构/训练过程可视化）

文章预览

点击上方 “ 小白学视觉 ”，选择加" 星标 "或“ 置顶 ” 重磅干货，第一时间送达 作者 | 锦恢@知乎（已授权）  来源 | https://zhuanlan.zhihu.com/p/220403674  编辑 | 极市平台  一、网络结构的可视化 我们训练神经网络时，除了随着step或者epoch观察损失函数的走势，从而建立对目前网络优化的基本认知外，也可以通过一些额外的可视化库来可视化我们的神经网络结构图。这将更加地高效地向读者展现目前的网络结构。 为了可视化神经网络，我们先建立一个简单的卷积层神经网络：   import  torch   import  torch.nn  as  nn     class   ConvNet (nn.Module) :       def   __init__ (self) :          super(ConvNet, self).__init__()            self.conv1 = nn.Sequential(              nn.Conv2d( 1 ,  16 ,  3 ,  1 ,  1 ),              nn.ReLU(),              nn.AvgPool2d( ………………………………