When I try to execute the code I get the following error: ValueError: tf.function-decorated function tried to create variables on non-first call. I am using tensorflow-gpu 2.0.0. Any suggestion?