diff --git a/climax/main/config.py b/climax/main/config.py index 1ca227b62f5fc8ed38d78de0d1f893b0bf1b830b..8d708e21453661d3038b2ce64e66c37a662a15b2 100644 --- a/climax/main/config.py +++ b/climax/main/config.py @@ -131,19 +131,19 @@ OPTIM = torch.optim.Adam BATCH_SIZE = 16 # maximum learning rate determined from learning rate range test -if PREDICTAND == 'tasmin': +if PREDICTAND is 'tasmin': if isinstance(LOSS, L1Loss): MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.004 if isinstance(LOSS, MSELoss): MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.002 -if PREDICTAND == 'tasmax': +if PREDICTAND is 'tasmax': if isinstance(LOSS, L1Loss): MAX_LR = 0.001 if isinstance(LOSS, MSELoss): MAX_LR = 0.001 if OPTIM is torch.optim.Adam else 0.004 -if PREDICTAND == 'pr': +if PREDICTAND is 'pr': if isinstance(LOSS, L1Loss): MAX_LR = 0.001 if isinstance(LOSS, MSELoss):