把torchsummary卸载掉
pip uninstall torchsummary
再重新安装个torch-summary
pip install torch-summary
上面的from torchsummary import summary
不用变。
把torchsummary卸载掉
pip uninstall torchsummary
再重新安装个torch-summary
pip install torch-summary
上面的from torchsummary import summary
不用变。