关于pytorch网络模型可视化函数make_dot的一些问题
首先,放上make_dot函数的源码(实验室同门给的,出处不详= =)
这个函数使用方法如下:
可能会出现的问题
problem one
emmm 还是会出现一些问题的= =,因为make_dot要用到graphviz,要注意graphviz的安装方法,不仅在实验环境上要安装,还要在系统里安装,否则就会出现graphviz.backend.ExecutableNotFound: failed to execute ['dot', '-Tpdf', '-O', 'Digraph.gv']graphviz.backend.,make sure the Graphviz executables are on your systems' PATH这样的错误([]里的内容可能不一样,但是问题是一样的,而且都会有'dot'),这里有一个博客总结的很好,————传送门———— 这里要加一句,安装好系统里的呵环境里的之后,一定要重启python内核,至于怎么做到我也不清楚,但是重启电脑一定可以(›´ω`‹ ) ,因为我就是重启电脑之后才work的~
problem two
之后可能还会出现这样的错误ImportError: cannot import name 'PILLOW_VERSION' from 'PIL',这个错误,就是pillow版本太高了,可以使用命令
或者小于7.0.0的版本都行,但是注意有时候,pip安装不上,那就用conda安装,conda不行就用pip,还有就是编辑器内部的命令行安装以及anaconda的prompt shell,总共有四种组合方式,分别如下:
1.在IDE内部的terminal用conda install
2.在IDE内部的terminal用pip install
3.在anaconda prompt shell 用conda install
4.在anaconda prompt shell 用pip install
有时候你安装好了还报错,就很有可能是没有安装上,这四种方式都要试一下,安装都能出bug,我也很无奈= =
problem three
后面还有可能出现一个问题就是graphviz.backend.CalledProcessError: Command '['dot', '-Tpdf', '-O', 'model_structure.pdf']' returned non-zero exit status 1. [stderr: b'Error: Could not open "modelstructure.pdf.pdf" for writing : Permission denied\r\n'] 类似这种错误,这个实际上是因为你多次调用make_dot,而调用make_dot会将画好的模型图打开,第二次调用就会报写错误,因为打开操作被拒绝,这个问题解决方法很简单,加上判断语句,只调用一次就行了~
就这么多啦,有什么疑问大家可以评论区交流呀!~( ˘•ω•˘ )~