From 41825b4b6ffa729b6f6428755eaf7ab690ba33d9 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 27 Oct 2021 10:14:41 +0200
Subject: [PATCH] Base learning rate for PR-only.
---
climax/main/config.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/climax/main/config.py b/climax/main/config.py
index b9c9c0d..67a1406 100644
--- a/climax/main/config.py
+++ b/climax/main/config.py
@@ -154,7 +154,7 @@ if PREDICTAND is 'pr':
if isinstance(LOSS, BernoulliGammaLoss):
# learning rates for supersampling task
if not any(ERA5_P_PREDICTORS) and ERA5_S_PREDICTORS == 'pr':
- MAX_LR = 0.002
+ MAX_LR = 0.001
else:
MAX_LR = 0.0005 if OPTIM is torch.optim.Adam else 0.001
--
GitLab