Model usage in Google Colab

[Geoka1]

Hello everyone,

I wanted to share a quick fix for those who are trying to use the model in Google Colab with the latest versions of TensorFlow.

Issue:
When using the the model in Colab with the newer TensorFlow versions, there are be compatibility issues due to the updated CUDA version that cannot be downgraded to cuda 11.2 anymore and GPU cannot be used. If you use the latest TensorFlow training fails because the optimizers cannot be loaded.

Solution:
You can switch 'tf.keras.optimizers.Adam' in models.py to 'tf.keras.optimizers.legacy.Adam' instead so that the model becomes compatible with the latest tensorflow version.

I hope this helps anyone facing similar issues.