当模型多输入时，torchsummary估计参数量

• 方法

方法

直接将参数传入即可。如下代码，有三个以上的输入也是以此类推

summary(model,first_input,second_input,device='cpu')

还有问题可以去这里查看torch-summary