@ptrblck Okay so that’s why these 2 lines are in five/ten-crop the documentation.
>>> result = model(input.view(-1, c, h, w)) # fuse batch size and ncrops
>>> result_avg = result.view(bs, ncrops, -1).mean(1) # avg over crops