diff --git a/ANN_Training.py b/ANN_Training.py
index 65dc77a772643b3354c886e78366fe59d1e62167..158c6f52972f08e8cdacadea34fdbc8cb784db1f 100644
--- a/ANN_Training.py
+++ b/ANN_Training.py
@@ -10,7 +10,7 @@ TODO: Fix random seed
 TODO: Write-protect all data and models
 TODO: Put legend outside plot (bbox_to_anchor)
 TODO: Put plotting into separate function
-TODO: Reduce number of testing epochs to 50
+TODO: Reduce number of testing epochs to 50 -> Done
 
 """
 import numpy as np
@@ -205,7 +205,7 @@ class ModelTrainer(object):
             Dictionary containing classification evaluation data.
 
         """
-        self.epoch_training(training_set, num_epochs=100, verbose=False)
+        self.epoch_training(training_set, num_epochs=50, verbose=False)
         self._model.eval()
 
         x_test, y_test = test_set