tensorflow.keras.fit uses more and more memory until crash

Question

I am trying to solve an exercise from a machine learning book, where a classifier should be trained on the cifar10 dataset using tensorflow and keras. I have attached a code