pytorch相关可视化的方式有好几种,这边选择尝试最简单的HiddenLayer:
参考:pytorch中使用hiddenlayer包+安装graphviz实现可视化网络_Chuckie今天也要学习!的博客-优快云博客_hiddenlayer 安装
1.安装graphviz模块
网上很多方式是直接去下载安装包安装再进行配置环境变量实现的,我这边参考博主,直接用conda来安装:conda install python-graphviz
奇怪,让我升级,还要安装这么多包?咱也不懂,就装呗
安装结束, 奇怪,以前是torch的1.12.1版本的,现在直接变成pytorch了,没有torch了,torchvision还降低了版本了,所幸还能之前,而且之前下面的的warning也没有了。。后面发现之前博主的环境是conda创建的pytorch1.7虚拟环境中进行的,所以自动升级了把
D:\ProgramData\Anaconda3\envs\pytorchenv\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
也不晓得 python-graphviz和graphviz啥关系,反正也装了。
2.安装hiddenlayer模块
直接使用pip install hiddenlayer
这个安装的好快,很小
3.测试
都安装完成了,测试一下,参考这篇:PyTorch学习笔记:训练过程的可视化(hiddenlayer库)_code_carrot的博客-优快云博客_hiddenlayer库
但是拷贝过来后报错,虚拟环境下执行命令安装库:conda install scikit-learn
安装完毕
重启下pycharm,没有报错了,再直接执行看看,又报错了
继续安装,安装结束重启下pycharm
太棒了,终于报缺少数据集错误了,等太久
改成true,再执行下,出来了,但是我电脑好像卡死了。。 电脑太差没办法。。
那最后试验算是成功咯!