diff --git a/training/training_scripts/NN_Filtering_HOP/training/cfg/stage1.json b/training/training_scripts/NN_Filtering_HOP/training/cfg/stage1.json
index 552926638ca268bd271b7b58f0dd061b8adfe41b..bff5c2ce38eb4eb42d4aaa4fca6cd409dfcf002b 100644
--- a/training/training_scripts/NN_Filtering_HOP/training/cfg/stage1.json
+++ b/training/training_scripts/NN_Filtering_HOP/training/cfg/stage1.json
@@ -4,16 +4,14 @@
     "stage1" : {
         "training" : {
             "//" : "the epoch to switch from L1 to L2 loss, default: 80, max_epochs: 100",
-            "mse_epoch" : 27,
-            "max_epochs" : 30,
+            "mse_epoch" : 37,
+            "max_epochs" : 40,
             "lr_scheduler" : {
                 "milestones" : [
-                    22,
-                    25
+                    30,
+                    34
                 ]
             }
         }
     }
-}
-
-           
+}
\ No newline at end of file