How to implement test time augmentation with pytorch lightning

abisoyeodurojaiye December 3, 2023, 5:38pm 1

I’ve read about test time augmentation but how would I implement this using say torchvision transforms/albumentations and the LightningModule?

Topic		Replies	Views
Pytorch Lightning Module not decreasing training loss/improving even training	0	1677	August 12, 2022
Loading weights to predict on train set LightningModule	0	414	August 5, 2021
Discrepancy between val and test metrics implementation help	6	3268	December 21, 2020
Implemenation of DETR implementation help	2	1359	September 22, 2020
Implement SCHEDULER OPTIMIZER in Pytorch Lightning implementation help	0	735	August 28, 2022