CancelledError: [_Derived_]RecvAsync is cancelled. Running on google colab

Question

I am getting the following error when fitting my model. Occurs after every 5th epoch!

Link to notebook

Tensorflow, Keras, Cuda versions

Error: