Parameters for torch summary

[BiaBibii]

Hello, I want to visualize all the layers of the model AARMVSNet with number of parameters, and I tried with torch summary, but I got some errors because of parameters, could you provide the parameters for summary(model, ........) ? I tried a lot of combinations, but it doesn't work.