Skip to content

Commit

Permalink
[Fix] Fix resume is not set correctly bug when cfg.resume is True (o…
Browse files Browse the repository at this point in the history
…pen-mmlab#221)

* add optimize_anchors for mmyolo

* update optimize_anchors

* optimize anchors script

* add parameter in args

* add optimize anchor related docs

* fix lint error

* fix docs

* change anchors output format

* fix docs error

* fix resume set

* fix resume mode set

* add mmeval relate docs

Co-authored-by: imAzhou <lyzhou@creams.io>
  • Loading branch information
2 people authored and hhaAndroid committed Nov 3, 2022
1 parent 275beec commit 188377c
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 3 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -246,3 +246,4 @@ This project is released under the [GPL 3.0 license](LICENSE).
- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab image and video editing toolbox.
- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox.
- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework.
- [MMEval](https://github.com/open-mmlab/mmeval): OpenMMLab machine learning evaluation library.
1 change: 1 addition & 0 deletions README_zh-CN.md
Original file line number Diff line number Diff line change
Expand Up @@ -249,6 +249,7 @@ MMYOLO 是一款由来自不同高校和企业的研发人员共同参与贡献
- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab 图像视频编辑工具箱
- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab 图片视频生成模型工具箱
- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab 模型部署框架
- [MMEval](https://github.com/open-mmlab/mmeval): OpenMMLab 机器学习算法评测库

## 欢迎加入 OpenMMLab 社区

Expand Down
2 changes: 1 addition & 1 deletion mmyolo/utils/setup_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ def register_all_modules(init_default_scope: bool = True):
Args:
init_default_scope (bool): Whether initialize the mmdet default scope.
When `init_default_scope=True`, the global default scope will be
set to `mmdet`, and all registries will build modules from mmdet's
set to `mmyolo`, and all registries will build modules from mmdet's
registry node. To understand more about the registry, please refer
to https://github.com/open-mmlab/mmengine/blob/main/docs/en/tutorials/registry.md
Defaults to True.
Expand Down
4 changes: 2 additions & 2 deletions tools/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,8 +86,8 @@ def main():
f'`OptimWrapper` but got {optim_wrapper}.')
cfg.optim_wrapper.type = 'AmpOptimWrapper'
cfg.optim_wrapper.loss_scale = 'dynamic'

cfg.resume = args.resume
if args.resume:
cfg.resume = args.resume

# build the runner from config
if 'runner_type' not in cfg:
Expand Down

0 comments on commit 188377c

Please sign in to comment.