# How to include Data Augmentation for Roboflow's SegFormer Colab Notebook?

**URL:** <https://discuss.roboflow.com/t/how-to-include-data-augmentation-for-roboflows-segformer-colab-notebook/836>\
**Category:** 🤝  Community Help\
**Tags:** bugs\
**Created:** [August 25, 2022, 1:16am UTC](https://discuss.roboflow.com/t/how-to-include-data-augmentation-for-roboflows-segformer-colab-notebook/836 "2022-08-25T01:16:17Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![zytoh0](https://yyz1.discourse-cdn.com/flex029/user_avatar/discuss.roboflow.com/zytoh0/32/695_2.png) [@zytoh0](https://discuss.roboflow.com/u/zytoh0)\
**Post date:** [August 25, 2022, 1:16am UTC](https://discuss.roboflow.com/t/how-to-include-data-augmentation-for-roboflows-segformer-colab-notebook/836/1 "2022-08-25T01:16:17Z")

</div>

Hi how can I include data augmentation in Roboflow’s Segformer guide.

I’ve tried following this guide with no avail: [GPU and batched data augmentation with Kornia and PyTorch-Lightning — PyTorch Lightning 2.1.0dev documentation](https://pytorch-lightning.readthedocs.io/en/latest/notebooks/lightning_examples/augmentation_kornia.html?highlight=data%20augmentation)

Can anyone show code additions/modifications to the SegFormer guide: [Google Colab](https://colab.research.google.com/drive/1250K828ixr-sG2xzLVYYN5AVa3cTaDEF?usp=sharing) required to make it work?