I was running some code in Google Colab. I defined my own model "MyModel()" and some functions (not shown because it\'s too long), which is inherited from \'tf.ker