From 9df9de3a2c9519c9ca670205aab4e6c41ea5e40b Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 21 Oct 2021 17:28:51 +0200
Subject: [PATCH] Setup for hyperparameter grid search.
---
climax/main/config.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/climax/main/config.py b/climax/main/config.py
index 1ca227b..8d708e2 100644
--- a/climax/main/config.py
+++ b/climax/main/config.py
@@ -131,19 +131,19 @@ OPTIM = torch.optim.Adam
BATCH_SIZE = 16
# maximum learning rate determined from learning rate range test
-if PREDICTAND == 'tasmin':
+if PREDICTAND is 'tasmin':
if isinstance(LOSS, L1Loss):
MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.004
if isinstance(LOSS, MSELoss):
MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.002
-if PREDICTAND == 'tasmax':
+if PREDICTAND is 'tasmax':
if isinstance(LOSS, L1Loss):
MAX_LR = 0.001
if isinstance(LOSS, MSELoss):
MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.004
-if PREDICTAND == 'pr':
+if PREDICTAND is 'pr':
if isinstance(LOSS, L1Loss):
MAX_LR = 0.001
if isinstance(LOSS, MSELoss):
--
GitLab