Added missing file

xEcho1337 · Jan 16, 2025 · 884777b · 884777b
1 parent 27efc9f
commit 884777b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/net/echo/brain4j/training/techniques/SmartTrainer.java b/src/main/java/net/echo/brain4j/training/techniques/SmartTrainer.java
@@ -45,7 +45,7 @@ public void start(Model model, DataSet dataSet, double lossThreshold) {
                 double finalLoss = loss;
                 listeners.forEach(listener -> listener.onEvaluated(dataSet, epoches, finalLoss));
 
-                if (loss > lastLoss) {
+                if (loss >= lastLoss) {
                     // Loss increased, so decrease the learning rate
                     model.getOptimizer().setLearningRate(model.getOptimizer().getLearningRate() * learningRateDecay);
                 }
@@ -79,7 +79,7 @@ public void startFor(Model model, DataSet dataSet, int epochesAmount) {
                 double finalLoss = loss;
                 listeners.forEach(listener -> listener.onEvaluated(dataSet, epoches, finalLoss));
 
-                if (loss > lastLoss) {
+                if (loss >= lastLoss) {
                     double finalLastLoss = lastLoss;
 
                     // Loss increased, so decrease the learning rate