mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Tong Gao 2022-03-12 00:21:22 +08:00 committed by GitHub
parent c4b02566cf
commit 00b003da23
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/runner/optimizer/default_constructor.py
View File

@ -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)