当模型多输入时,torchsummary估计参数量
方法
直接将参数传入即可。如下代码,有三个以上的输入也是以此类推
summary(model,[first_input,second_input],device='cpu')
如果还是报错的话,pip list 看看是不是装错库了。
python中有torchsummary 和torch_summary两个库。其中一个功能区别就是一个支持多输入,一个不支持。
还有问题可以去这里查看torch-summary
感谢@DebinRhode指正。前面少写了括号。