You can refer to the following URL
https://github.com/tensorflow/tensorflow/issues/86953#event-16275455512
This seems to be a problem with keras
I used this method to solve it before
You can try it
But in recent days, colab TPU seems to have problems and I can't connect to TPU