Skip to content

GitLab

Unverified Commit f6d5b0f8 authored by Cao Yuhang's avatar Cao Yuhang Committed by GitHub
Browse files

fix lint #580 (#590)

parent c80e4cae
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
mmcv/runner/hooks/optimizer.py
View file @ f6d5b0f8
...@@ -73,9 +73,12 @@ class Fp16OptimizerHook(OptimizerHook): ...@@ -73,9 +73,12 @@ class Fp16OptimizerHook(OptimizerHook):
runner.optimizer.param_groups = copy.deepcopy( runner.optimizer.param_groups = copy.deepcopy(
runner.optimizer.param_groups) runner.optimizer.param_groups)
state = defaultdict(dict) state = defaultdict(dict)
p_map = {old_p: p for old_p, p in p_map = {
zip(chain(*(g['params'] for g in old_groups)), old_p: p
chain(*(g['params'] for g in runner.optimizer.param_groups)))} 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(): for k, v in runner.optimizer.state.items():
state[p_map[k]] = v state[p_map[k]] = v
runner.optimizer.state = state runner.optimizer.state = state
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment