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}_valloss{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