corrected restoring of saved pars

supsi-dacd-isaac · Dec 14, 2023 · f9b9105 · f9b9105
1 parent d55e6a8
commit f9b9105
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyforecaster/forecasting_models/neural_forecasters.py b/pyforecaster/forecasting_models/neural_forecasters.py
@@ -315,10 +315,10 @@ def fit(self, inputs, targets, n_epochs=None, savepath_tr_plots=None, stats_step
                 k += 1
             if finished:
                 break
-            if len(val_loss)>1:
-                if val_loss[-1] > val_loss[-2]:
-                    pars = old_pars
-            self.pars = pars
+        if len(val_loss)>2:
+            if val_loss[-1] > val_loss[-2]:
+                pars = old_pars
+        self.pars = pars
         super().fit(inputs_val_0, targets_val_0)
         return self