Skip to content

GitLab

Unverified Commit 00b003da authored by Tong Gao's avatar Tong Gao Committed by GitHub
Browse files

[Docs] Fix a wrong example in DefaultOptimizerConstructor (#1795)

parent c4b02566
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/runner/optimizer/default_constructor.py
View file @ 00b003da
...@@ -84,7 +84,7 @@ class DefaultOptimizerConstructor: ...@@ -84,7 +84,7 @@ class DefaultOptimizerConstructor:
>>> # assume model have attribute model.backbone and model.cls_head >>> # assume model have attribute model.backbone and model.cls_head
>>> optimizer_cfg = dict(type='SGD', lr=0.01, weight_decay=0.95) >>> optimizer_cfg = dict(type='SGD', lr=0.01, weight_decay=0.95)
>>> paramwise_cfg = dict(custom_keys={ >>> paramwise_cfg = dict(custom_keys={
'.backbone': dict(lr_mult=0.1, decay_mult=0.9)}) 'backbone': dict(lr_mult=0.1, decay_mult=0.9)})
>>> optim_builder = DefaultOptimizerConstructor( >>> optim_builder = DefaultOptimizerConstructor(
>>> optimizer_cfg, paramwise_cfg) >>> optimizer_cfg, paramwise_cfg)
>>> optimizer = optim_builder(model) >>> optimizer = optim_builder(model)
......
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