I am using Kaggle platform to run this code which have tensorflow 2.6 and keras 2.6. And I found it runs for epoch 1 and stuck somewhere in custom generator.

[sneh-debug]

Also in custom_generator script. I used on_epoch_end(self) rather than self.on_epoch_end(). Please suggest what can be the reasons?