Base learning rate for PR-only.

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