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

Question

我正在使用手电筒摘要

from torchsummary import summary

我想在打印模型摘要时传递多个参数，但这里提到的示例: Model summary in pytorch只采取了一个论点。例如:

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

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

def forward(self, img1, img2):

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

Answer 1

您可以使用此处给出的示例:pytorch summary multiple inputs

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