torchinfo
torchinfo copied to clipboard
multi-parameters forward function doesnt work
Hello, first of all thank you for your amazing work.
Describe the bug I have a model with the function forward take x and y in parameters.
...
def forward(x, y):
pred = model(x,y)
...
What part of the code should I rewrite to take this modification into account ?
Try using explicit kwargs:
summary(model, x=x, y=y)
Or using a tuple for input data:
summary(model, input_data=(x,y)