Как получить сводную информацию о модели PyTorch, которая использует словарь в качестве входных данных ⇐ Python

[Anonymous]

Моя модель принимает в качестве входных данных словарь, например x = {'image': torch.tensor, 'number': torch.tensor}, модель выглядит так:
class MyModel(nn.Module):
def __init__(self):
super(MyModel, self).__init__()
self.imgmodule = ImgModule()
self.nummodule = NumModule()
self.predict = nn.Linear(input_size, 100)

def forward(self, x):
xImg = self.imgmodule(x['image'])
xNum = self.nummodule(x['number'])
x = self.predict(torch.cat([xImg, xNum], dim=1))
return x

Как получить сводку модели, аналогичную сводке, представленной в пакете pytorch-summary?
До сих пор я пробовал использовать ее следующим образом :
from torchsummary import summary
summary(model, input_size=[(3, 224, 224), (1, )])

Но я получаю сообщение об ошибке:
TypeError: MyModel.forward() takes 2 positional arguments but 3 were given


Подробнее здесь: https://stackoverflow.com/questions/791 ... s-an-input