Implemented early stopping

tcotte · Dec 21, 2017 · b8610d8 · b8610d8
1 parent 7c5f2e6
commit b8610d8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -273,6 +273,10 @@ def train_nn(sess, epochs, data_folder, image_shape, batch_size, training_image_
         training_loss_metrics.append(training_loss)
         training_accuracy_metrics.append(training_accuracy)
 
+        if epoch > 0 and (training_accuracy_metrics[-1] < training_accuracy_metrics[-2] - 0.01):
+            print("Early stopping!!!!")
+            break
+
         print(
             "Epoch %d:" % (epoch + 1),
             "Training loss: %.4f, accuracy: %.2f" % (training_loss, training_accuracy),