Failed to run model on torch 1.10+cuda 11.3

[davimedio01]

When I test the model with the torchinfo library, i.e. by creating some input with a size = (1, 3, 16, 224, 224) to simulating my K400 dataset entry and sending it to the model, the following error occurs: By the way, the model is composed with a backbone (TPS - SwinTransformer2D_TPS) and a head (I3DHead), connected by a torch.nn.Sequential. torch 1.10.0 // cuda 11.3 // ubuntu 22.04 wsl