DEEPSTORM GOOGLE COLAB - issue with training the model

[h122233334]

When going to train the DEEPSTORM (2D) model, I am running into issues with the system RAM usage and the coding. It tells me the file save format cannot be hdf5 and must be in keras format. After I change the code from hdf5 to keras there is either an issue with the training taking up too much system RAM or there is another issue with the code.

To Reproduce
Steps to reproduce the behavior:

1. Go to 4.4. Start Training
2. Click on run cell
3. See error

Expected behavior
I expect the cell to run without any errors and to get a trained model.

Screenshots

Desktop (please complete the following information):

• OS: MacOS Sequoia 15.2
• Browser Safari
• Version 18.2

Add any other context about the problem here.