From 4ba9390b6dd91b012ad0f7e541ee1c856c55bb46 Mon Sep 17 00:00:00 2001
From: Franck Galpin <franck.galpin@interdigital.com>
Date: Wed, 6 Sep 2023 11:44:13 +0000
Subject: [PATCH] change stage 2 nb epochs from 20 to 19 to match what was done
during HOP training
---
training/training_scripts/NN_Filtering_HOP/default_config.json | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/training/training_scripts/NN_Filtering_HOP/default_config.json b/training/training_scripts/NN_Filtering_HOP/default_config.json
index a1820e611..8bdff2690 100644
--- a/training/training_scripts/NN_Filtering_HOP/default_config.json
+++ b/training/training_scripts/NN_Filtering_HOP/default_config.json
@@ -330,7 +330,7 @@
"multi_gpu": false,
"validation_interval": 1,
"mse_epoch": 18,
- "max_epochs": 20,
+ "max_epochs": 19,
"component_loss_weightings": [
6,
2
--
GitLab