Solved by Oli in post #7 
              
              
                Just curious here. What’s the benefit of doing another forward pass on the result? 
result = self.model(input_var.view(-1, c, h, w)) # fuse batch size and ncrops
output = self.model(result)