How to include Data Augmentation for Roboflow's SegFormer guide?

Hi how can I include data augmentation in Roboflow’s Segformer guide: Google Colab

I’ve tried following this guide with no avail: GPU and batched data augmentation with Kornia and PyTorch-Lightning — PyTorch Lightning 1.8.0dev documentation

Can anyone show code additions/modifications to the SegFormer guide required to make it work?

Hello - our automated system accidently flagged this message as spam. Could you please post again and we can assist? Thank you.

Hi I’ve reposted here: How to include Data Augmentation for Roboflow's SegFormer Colab Notebook?