How to run torchinfo on 2 inputs ?

[amitli1]

I have a torch model which get's 2 inputs:
(batch_size, 90, 4) and (batch_size)

I run the model as follows:

# x has the shape (batch_size, 90, 4)
# s had the shape (batch_size)
res = model(x, s)



How to run torchinfo.summary function  ? 

[TylerYep]

Something like:

from torchinfo import summary

results = summary(model, input_data=[x, s])