记录一下自己在Pytorch学习中,Tensorboard可视化网络的过程。
1. 环境配置
在可视化过程中,刚开始环境为tensorflow==2.0.1,tensorboardX==1.9,运行代码会出现如下的问题:
AttributeError: 'torch._C.Value' object has no attribute 'debugName'。
根据https://blog.csdn.net/u012702061/article/details/102870371的建议,更换到如下版本:
Python |
|
3.5.6 :: Anaconda |
torch |