call ModelCheckpoint

backend/IT18218640/train.py

@@ -109,14 +109,14 @@ def main():
     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))
+    history = model.fit(X_train, y_train, epochs=EPOCHS, batch_size=BATCH_SIZE, validation_data=(X_validation, y_validation), callbacks=[checkpoint])
 
     # evaluate the model
     test_error, test_accuracy = model.evaluate(X_test, y_test)
     print(f"Test error: {test_error}, test accuracy: {test_accuracy}")
 
     # save the model
-    model.save(SAVE_MODEL_PATH)
+    # model.save(SAVE_MODEL_PATH)
 
     show_graph(history)