1. 出现问题
在使用torchsummary对pytorch网络结构可视化时,出现如下错误:

最秀的是,运行完summary后,让模型对输入进行处理也会出现该错误:

找了半天bug,发现原来是包安装错了
2. 解决问题
将torchsummary卸载后,安装torch-summary
pip uninstall torchsummary
pip install torch-summary
输出网络结构:

这个bug真是有点迷
本文记录了在使用torchsummary进行PyTorch网络结构可视化时遇到的问题及解决方案。主要原因是包安装错误,通过卸载torchsummary并安装torch-summary解决了问题。
在使用torchsummary对pytorch网络结构可视化时,出现如下错误:

最秀的是,运行完summary后,让模型对输入进行处理也会出现该错误:

找了半天bug,发现原来是包安装错了
将torchsummary卸载后,安装torch-summary
pip uninstall torchsummary
pip install torch-summary
输出网络结构:

这个bug真是有点迷
您可能感兴趣的与本文相关的镜像
PyTorch 2.8
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
5751
3593
1931