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
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 atorch.nn.Sequential.torch 1.10.0 // cuda 11.3 // ubuntu 22.04 wsl