Hi @Stefano_Di_Pietro , I am training the model with GPU enabled in Colab (in the cloud) without issues (notebook: Google Colaboratory 2 → Edit > Notebook settings > GPU).
So, this may be specific to your setup. Have you had issues with training other TensorFlow 2.x models on your GPU? Can you share your OS, GPU model, cuDNN/CUDA versions?
Hardware requirements: GPU support  |  TensorFlow 2
Software requirements: GPU support  |  TensorFlow 2
GPU support: GPU support  |  TensorFlow