Use of GPU on Colab

[fabiox23]

Hi,

I have tried running the script on Colab following the instructions on the README file and installing the missing dependencies.

It does work but it doesn't seem to use the GPU despite the option being set up.

How can I ensure the GPU is used? Is it something to do with tensorflow? Any advice?

Kind regards.