1. 出现问题
在使用torchsummary对pytorch网络结构可视化时,出现如下错误:
最秀的是,运行完summary后,让模型对输入进行处理也会出现该错误:
找了半天bug,发现原来是包安装错了
2. 解决问题
将torchsummary卸载后,安装torch-summary
pip uninstall torchsummary
pip install torch-summary
输出网络结构:
这个bug真是有点迷
在使用torchsummary对pytorch网络结构可视化时,出现如下错误:
最秀的是,运行完summary后,让模型对输入进行处理也会出现该错误:
找了半天bug,发现原来是包安装错了
将torchsummary卸载后,安装torch-summary
pip uninstall torchsummary
pip install torch-summary
输出网络结构:
这个bug真是有点迷