Solved by ptrblck in post #2 
              
              
                nn.Conv3d expects an input in the shape [batch_size, channels, depth, height, width], so you would need to reshape the output of the pretrained model to this shape. 
Depending which model you are using, it might already provide the desired shape or you would need to figure out how this should be don…