Сводка модели pytorch - у foward более одного аргумента

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(model, [(1, 16, 16), (1, 28, 28)])