Perform Test time Augmentation?

[shreyaskamathkm]

Hi,

Thank you for providing such detailed example for semantic segmentation with PyTorch lightning. I was wondering if you have any idea abut implementing test time augmentation in the validation loop and maybe save the images as well? Any help is appreciated.

Thank you, Best, SK