Skip to content

Commit

Permalink
[Doc]: Update config.md (open-mmlab#562)
Browse files Browse the repository at this point in the history
* Update config.md

* Update config.md
  • Loading branch information
Zhengfei-0311 authored and austinmw committed Oct 8, 2022
1 parent c0c0c6e commit 3da4e04
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/zh_cn/tutorials/config.md
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ optimizer = dict(type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0001)
然后在算法库中可以通过如下代码构造优化器对象。

```python
from mmengine import Config
from mmengine import Config, optim
from mmengine.registry import OPTIMIZERS

import torch.nn as nn
Expand Down Expand Up @@ -573,7 +573,7 @@ MMEngine 还提供了 `get_config` 和 `get_model` 两个接口,支持对符
用户可以通过指定 `pretrained=True` 获得已经加载预训练权重的模型以进行训练或者推理。

```python
from mmengine import get_model
from mmengine.hub import get_model

model = get_model(
'mmdet::faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py', pretrained=True)
Expand All @@ -590,7 +590,7 @@ http loads checkpoint from path: https://download.openmmlab.com/mmdetection/v2.0
同时,如果用户指定 `pretrained=True` ,得到的配置文件中会新增 `model_path` 字段,指定了对应模型预训练权重的路径。

```python
from mmengine import get_config
from mmengine.hub import get_config

cfg = get_config(
'mmdet::faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py', pretrained=True)
Expand Down

0 comments on commit 3da4e04

Please sign in to comment.