keras-checkpoint.py

from tensorflow.keras.callbacks import ModelCheckpoint
checkpoint_path  = "/opt/ml/checkpoints"
checkpoint_names = 'cifar10-'+model_type+'.{epoch:03d}.h5'
checkpoint_callback = ModelCheckpoint(filepath=f'{checkpoint_path}/{checkpoint_names}',
                                          save_weights_only=False,
                                          monitor='val_loss')

model.fit(train_dataset, ...
          epochs=epochs,
          initial_epoch=epoch_number,
          callbacks=[checkpoint_callback])