Setup for hyperparameter grid search.

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):