Chapter 7 - Further research Implementing TTA

Part 1 (2020)
10

From what I can tell in the code, you would need to pass all the item transforms you want, as it overrides it. So IE:

item_tfms = [RandomResizedCrop(mysize), ToTensor()]

Or

batch_tfms = [RandomResizedCropGPU(mysize), IntToFloatTensor(), Normalize.from_stats(*imagenet_stats)]
1 Like