1 files changed, 1 insertions, 0 deletions
diff --git a/lenet.py b/lenet.py
index cae1afa..3ddab06 100644
--- a/lenet.py
+++ b/lenet.py
@@ -117,6 +117,7 @@ def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100,
   if keep_training:
     model.load_weights('./weights.h5')
   history = model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, verbose=1, validation_data = (x_val, y_val))
+  model.save_weights('./model_gan.h5')
   plot_history(history, 'categorical_accuracy')
   plot_history(history)
   model.save_weights('./weights.h5')