fix lint #580 (#590)

open-mmlab · Sep 28, 2020 · f6d5b0f · f6d5b0f
1 parent c80e4ca
commit f6d5b0f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mmcv/runner/hooks/optimizer.py b/mmcv/runner/hooks/optimizer.py
@@ -73,9 +73,12 @@ def before_run(self, runner):
         runner.optimizer.param_groups = copy.deepcopy(
             runner.optimizer.param_groups)
         state = defaultdict(dict)
-        p_map = {old_p: p for old_p, p in
-                 zip(chain(*(g['params'] for g in old_groups)),
-                     chain(*(g['params'] for g in runner.optimizer.param_groups)))}
+        p_map = {
+            old_p: p
+            for old_p, p in zip(
+                chain(*(g['params'] for g in old_groups)),
+                chain(*(g['params'] for g in runner.optimizer.param_groups)))
+        }
         for k, v in runner.optimizer.state.items():
             state[p_map[k]] = v
         runner.optimizer.state = state