tensorwatch 可视化pytorch 网络结构

时间:2024-03-23 15:19:02

1.安装tensorwatch

pip install tensorwatch

2.安装后调用发现会有很多错误,包括但不限于


2.1 AttributeError: module 'torch.onnx' has no attribute 'set_training'

原因:pytorch版本太高,我是1.6,而1.6以下的版本torch.onnxhas 才有属性 'set_training'

办法:把pytorch的版本降低,可以直接使用

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2

2.2 AttributeError: ‘Dot’ object has no attribute ‘repr_svg’ 

原因:也是版本问题

办法:Anaconda\Lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py的第13行改为 return self.dot.create_svg().decode()

2.3 FileNotFoundError: [Error 2] "dot" not found in path

原因:Graphviz和pydot的问题

办法:点击链接

我装好Graphviz就没有问题了。

大部分问题都是版本不匹配问题。

更多安装内容参考: https://blog.csdn.net/qq_35878757/article/details/103561923

3.画网络结构图demo

tensorwatch 可视化pytorch 网络结构

存在的问题:用pycharm就不出图,使用jupyter notebook才出图