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.

Mehrdad-Noori / Brain-Tumor-Segmentation

Attention-Guided Version of 2D UNet for Automatic Brain Tumor Segmentation

207 stars 37 forks source link

Open sneh-debug opened 1 year ago

sneh-debug commented 1 year ago

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