Failed to train UNET

[jinxsfe]

Hi, adjust UNET structure for 4 depth, instead of 3 depth, and training show an error, for weight names, can you give me the sign how to adjust original coding part? I did not find the part for this error shown

ValueError Traceback (most recent call last)
in <cell line: 18>()
16 start = time.time()
17 # Start Training
---> 18 model.train(epochs=number_of_epochs,
19 batch_size=batch_size,
20 train_generator=train_generator,

1 frames
/usr/local/lib/python3.10/dist-packages/keras/src/callbacks/model_checkpoint.py in init(self, filepath, monitor, verbose, save_best_only, save_weights_only, mode, save_freq, initial_value_threshold)
181 if save_weights_only:
182 if not self.filepath.endswith(".weights.h5"):
--> 183 raise ValueError(
184 "When using save_weights_only=True in ModelCheckpoint"
185 ", the filepath provided must end in .weights.h5 "

ValueError: When using save_weights_only=True in ModelCheckpoint, the filepath provided must end in .weights.h5 (Keras weights format). Received: filepath=/content/gdrive/MyDrive/cryo/cryo-data_processing_volume/model/gaussian4layer_320_50_0.00014/ckpt/gaussian4layer_320_50_0.00014.hdf5

[jinxsfe]

the save weight only had set false but error is stills shown

[jinxsfe]

@esgomezm

[jinxsfe]

hi, I tried the new original version notebook from your website, and also met the same issue, can you help me to check with that?

@esgomezm

[jinxsfe]

when I set parameter, it seems need to change hdf5 to keras, and when I change it, the new error arising,.

[jinxsfe]

ckpt_name = ckpt_dir + '/' + model_name + 'epoch{epoch:02d}val_loss{val_loss:.4f}.weights.h5,
When I set it save_weight= True, and set weights,h5 in end of ckpt_name

the same error occur

[HalfdanJ]

Just ran into this same issue (and another issue). The keras version installed by default on colab is keras 3, and i think this only supports keras 2. I downgraded by running ! pip install keras==2.15.0 tensorflow==2.15.1, and that worked.

[jinxsfe]

Just ran into this same issue (and another issue). The keras version installed by default on colab is keras 3, and i think this only supports keras 2. I downgraded by running ! pip install keras==2.15.0 tensorflow==2.15.1, and that worked.

thank you!