Out of memory when using transforms

Hello,

I’m trying to do some data augmentation using transforms with torchaudio during training step.
After like 10 steps I get an “Out of memory” error.

I’ve tried torch.no_grad() but still the same. How to do transformations without getting this error ?
I’m kinda new so I’m sure that I’m missing something.

Thank you for help !