mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
parent
a5eb179153
commit
4e5c40b51b
@ -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
|
||||
@ -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)
|
||||
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user