Setup for hyperparameter grid search.

climax/main/config.py

@@ -155,7 +155,9 @@ if PREDICTAND is 'pr':
 BASE_LR = MAX_LR / 4
 
 # optimization parameters
-OPTIM_PARAMS = {'lr': BASE_LR, 'weight_decay': 0}
+OPTIM_PARAMS = {'lr': BASE_LR,
+                'weight_decay': 0
+                }
 if OPTIM is torch.optim.SGD:
     OPTIM_PARAMS['momentum'] = 0.99  # SGD with momentum