本文记录了在使用torchsummary进行PyTorch网络结构可视化时遇到的问题及解决方案。主要原因是包安装错误，通过卸载torchsummary并安装torch-summary解决了问题。
1. 出现问题
在使用torchsummary对pytorch网络结构可视化时，出现如下错误：
最秀的是，运行完summary后，让模型对输入进行处理也会出现该错误：
找了半天bug，发现原来是包安装错了
2. 解决问题
将torchsummary卸载后，安装torch-summary
pip uninstall torchsummary
pip install torch-summary
输出网络结构：
这个bug真是有点迷
扫一扫
评论 2
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
