Solved by Oli in post #2 
              
              
                Hello! You probably forgot to put the model in eval() mode. train() mode and batch size of 1 doesn’t make sense for resnet that uses batch norm