create model folder and ModelCheckpoint

backend/IT18218640/train.py

@@ -105,6 +105,9 @@ def main():
     input_shape = (X_train.shape[1], X_train.shape[2], X_train.shape[3],)
     model = build_model(input_shape, LEARNING_RATE)
 
+    # check Point
+    checkpoint = ModelCheckpoint('models/model-{epoch:03d}.h5', monitor='val_loss', save_best_only=True, mode='auto')
+
     # train the model
     history = model.fit(X_train, y_train, epochs=EPOCHS, batch_size=BATCH_SIZE, validation_data=(X_validation, y_validation))