From 6dc99665df2f58789640d244e63ad5f6b304b66c Mon Sep 17 00:00:00 2001
From: Franck Galpin <franck.galpin@interdigital.com>
Date: Thu, 29 Jun 2023 17:41:07 +0200
Subject: [PATCH] put back 40 epochs stage1 file

---
 .../NN_Filtering_HOP/training/cfg/stage1.json        | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

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 552926638c..bff5c2ce38 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
-- 
GitLab