Multiple input shape for torchinfo.summary

sivannavis: <p>Hi! I’m trying to see the summary of an audio-visual model. It’s a model wrapper that takes in audio and video input separately. But when I run <code>summary(model, [(1,8,3,112,112), (1, 4, 512, 128)])</code> it shows error <code>RuntimeError: Failed to run torchinfo. See above stack traces for more details. Executed layers up to: []</code>. What should I do?</p>
<p>My model looks like this without specifying the input shapes:</p>
<pre><code class="lang-auto">=================================================================
Layer (type:depth-idx)                   Param #
=================================================================
AV_Wrapper                               --
├─R2Plus1D: 1-1                          --
│    └─Sequential: 2-1                   --
│    │    └─Conv3d: 3-1                  28,224
│    │    └─BatchNorm3d: 3-2             128
│    │    └─ReLU: 3-3                    --
│    └─Sequential: 2-2                   --
│    │    └─BasicR2P1DBlock: 3-4         98,816
│    │    └─BasicR2P1DBlock: 3-5         98,816
│    └─Sequential: 2-3                   --
│    │    └─BasicR2P1DBlock: 3-6         328,704
│    │    └─BasicR2P1DBlock: 3-7         394,240
│    └─Sequential: 2-4                   --
│    │    └─BasicR2P1DBlock: 3-8         1,312,768
│    │    └─BasicR2P1DBlock: 3-9         1,574,912
│    └─Sequential: 2-5                   --
│    │    └─BasicR2P1DBlock: 3-10        5,246,976
│    │    └─BasicR2P1DBlock: 3-11        6,295,552
│    └─AdaptiveMaxPool3d: 2-6            --
├─Conv2D: 1-2                            --
│    └─Sequential: 2-7                   --
│    │    └─Conv2d: 3-12                 12,544
│    │    └─BatchNorm2d: 3-13            128
│    │    └─ReLU: 3-14                   --
│    └─Basic2DBlock: 2-8                 --
│    │    └─Conv2d: 3-15                 36,864
│    │    └─BatchNorm2d: 3-16            128
│    │    └─Conv2d: 3-17                 36,864
│    │    └─BatchNorm2d: 3-18            128
│    │    └─ReLU: 3-19                   --
│    └─Basic2DBlock: 2-9                 --
│    │    └─Conv2d: 3-20                 73,728
│    │    └─BatchNorm2d: 3-21            256
│    │    └─Conv2d: 3-22                 147,456
│    │    └─BatchNorm2d: 3-23            256
│    │    └─ReLU: 3-24                   --
│    └─Basic2DBlock: 2-10                --
│    │    └─Conv2d: 3-25                 294,912
│    │    └─BatchNorm2d: 3-26            512
│    │    └─Conv2d: 3-27                 589,824
│    │    └─BatchNorm2d: 3-28            512
│    │    └─ReLU: 3-29                   --
│    └─Basic2DBlock: 2-11                --
│    │    └─Conv2d: 3-30                 1,179,648
│    │    └─BatchNorm2d: 3-31            1,024
│    │    └─Conv2d: 3-32                 2,359,296
│    │    └─BatchNorm2d: 3-33            1,024
│    │    └─ReLU: 3-34                   --
│    └─AdaptiveMaxPool2d: 2-12           --
=================================================================
Total params: 20,114,240
Trainable params: 20,114,240
Non-trainable params: 0
=================================================================
</code></pre>

ptrblck: <p>Could you post a minimal and executable code snippet reproducing the issue?</p>