From d083154c2defe900d446ee9861b1fc887aa462fb Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 21 Oct 2021 17:32:35 +0200
Subject: [PATCH] Setup for hyperparameter grid search.
---
climax/main/config.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/climax/main/config.py b/climax/main/config.py
index 8d708e2..56fcbce 100644
--- a/climax/main/config.py
+++ b/climax/main/config.py
@@ -155,7 +155,9 @@ if PREDICTAND is 'pr':
BASE_LR = MAX_LR / 4
# optimization parameters
-OPTIM_PARAMS = {'lr': BASE_LR, 'weight_decay': 0}
+OPTIM_PARAMS = {'lr': BASE_LR,
+ 'weight_decay': 0
+ }
if OPTIM is torch.optim.SGD:
OPTIM_PARAMS['momentum'] = 0.99 # SGD with momentum
--
GitLab