Droid786:

Is there a faster way to achieve that.
I am using FiveCrop for size 224 but it is taking 35 minutes for 1 epoch.


Is FiveCrop slowing down the code and if so how long is an epoch taking without this transformation, i.e. for a single input only?



 Droid786:

Also, how the testing will pan out in this case. Right now, I am doing the RandomCrop of the same 224 size?


I’m not familiar with your use case and thus asked, if you could describe the general use case a bit.
E.g. if the image should get a single prediction, you could use the majority prediction of all 5 patches.
If all predictions are different, you could try to use the prediction with the highest logit.