How can I run the program on GPU instead of CPU?

[JucyCherry]

I found it was kind of slow when do SpecAugment on the CPU; and I would like to know which device you run your SpecAugment program on, CPU or GPU? If the SpecAugment program can be run using GPU, then how can I run the program using GPU instead of CPU?

[Luux]

I tried a bit around and had the same problem. I played with a Google Colab Notebook and the tf implementation was painfully and unusable slow there :/

[MichelleYang2017]

Both Slow. I am so sad

[AASHISHAG]

@JucyCherry : Did you find a way to run the code on GPU?