fix lint #580 (#590)

f6d5b0f8 · Cao Yuhang · GitHub · c80e4cae · f6d5b0f8
Unverified Commit f6d5b0f8 authored Sep 28, 2020 by Cao Yuhang Committed by GitHub Sep 28, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

mmcv/runner/hooks/optimizer.py mmcv/runner/hooks/optimizer.py +6 -3

No files found.
--- a/mmcv/runner/hooks/optimizer.py
+++ b/mmcv/runner/hooks/optimizer.py
@@ -73,9 +73,12 @@ class Fp16OptimizerHook(OptimizerHook):
        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