diff --git a/climax/main/config.py b/climax/main/config.py
index 8d708e21453661d3038b2ce64e66c37a662a15b2..56fcbceefabb7e7c531fc70bcb017c3e2ff7f144 100644
--- a/climax/main/config.py
+++ b/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