Set base LR for PR-supersampling task.

climax/main/config.py

@@ -149,7 +149,11 @@ if PREDICTAND is 'pr':
     if isinstance(LOSS, MSELoss):
         MAX_LR = 0.0004
     if isinstance(LOSS, BernoulliGammaLoss):
-        MAX_LR = 0.0005 if OPTIM is torch.optim.Adam else 0.001
+        # learning rates for supersampling task
+        if not any(ERA5_P_PREDICTORS) and ERA5_S_PREDICTORS == 'pr':
+            MAX_LR = 0.002
+        else:
+            MAX_LR = 0.0005 if OPTIM is torch.optim.Adam else 0.001
 
 # base learning rate: MAX_LR / 4 (Smith L. (2017))
 BASE_LR = MAX_LR / 4