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