Skip to content
Snippets Groups Projects
Commit 41825b4b authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Base learning rate for PR-only.

parent 6445ee9d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
climax/main/config.py
+ 1
− 1
View file @ 41825b4b
...@@ -154,7 +154,7 @@ if PREDICTAND is 'pr': ...@@ -154,7 +154,7 @@ if PREDICTAND is 'pr':
if isinstance(LOSS, BernoulliGammaLoss): if isinstance(LOSS, BernoulliGammaLoss):
# learning rates for supersampling task # learning rates for supersampling task
if not any(ERA5_P_PREDICTORS) and ERA5_S_PREDICTORS == 'pr': if not any(ERA5_P_PREDICTORS) and ERA5_S_PREDICTORS == 'pr':
MAX_LR = 0.002 MAX_LR = 0.001
else: else:
MAX_LR = 0.0005 if OPTIM is torch.optim.Adam else 0.001 MAX_LR = 0.0005 if OPTIM is torch.optim.Adam else 0.001
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment