Layer count mismatch when loading weights from file

[bastianspirek]

When running main.py I get the following error:

ValueError: Layer count mismatch when loading weights from file. Model expected 29 layers, found 30 saved layers.

I checked if I can reproduce the error when removing all other code and simply trying to load the model, and it still occurs:

from keras.models import load_model
from utils import GlobalWeightedAveragePooling2D

model_filepath = 'Melnyk-Net.hdf5'

model = load_model(model_filepath, custom_objects={"GlobalWeightedAveragePooling2D": GlobalWeightedAveragePooling2D})
model.summary()

I'm running Python 3.12. Thanks for the help!