diff --git a/climax/main/config.py b/climax/main/config.py index b9c9c0d6bd62be6f23c989b2d504b4f886bd1528..67a1406c9704c26e44908f3687ec29d28573fd83 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