pytorch 模型摘要 - forward func 有多个参数

Question

我正在使用手电筒摘要

from torchsummary import summary

我想在打印模型摘要时传递多个参数，但是这里提到的示例：pytorch中的模型摘要仅采用了一个参数。例如：

model = Network().to(device)
summary(model,(1,28,28))

原因是 forward 函数需要两个参数作为输入，例如：

def forward(self, img1, img2):

我如何在这里传递两个参数？

Answer 1

您可以使用此处给出的示例：pytorch summary multiple input

summary(model, [(1, 16, 16), (1, 28, 28)])