Doesn't work anymore

[Zumbalamambo]

It fails when it accessed GPU. Even after reducing the batch size in colab, it still fails.

If I try to train in CPU, it throws the following error,

Epoch 1/4
2020-06-06 23:16:08.769001: W tensorflow/core/framework/op_kernel.cc:1753] OP_REQUIRES failed at resource_variable_ops.cc:541 : Not found: Resource localhost/_AnonymousVar117/class tensorflow::Var does not exist.
2020-06-06 23:16:09.299689: W tensorflow/core/framework/op_kernel.cc:1753] OP_REQUIRES failed at resource_variable_ops.cc:541 : Not found: Resource localhost/_AnonymousVar118/class tensorflow::Var does not exist.

I'm using tensorflow version 2.3