[Feature] Resume from the latest checkpoint automatically. #6727

Czm369 · 2021-12-08T09:48:48Z

Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.

Motivation

Resume from the latest checkpoint automatically without specifying the path of the checkpoint.

Modification

Add a new argument auto-resume in tools/train.py and add a new function find_latest_checkpoint in mmdet/utils/misc.py.

BC-breaking (Optional)

Does the modification introduce changes that break the backward-compatibility of the downstream repos?
If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.

Use cases (Optional)

If this PR introduces a new feature, it is better to list some use cases here, and update the documentation.

Checklist

Pre-commit or other linting tools are used to fix the potential lint issues.
The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
If the modification has potential influence on downstream projects, this PR should be tested with downstream projects, like MMDet or MMCls.
The documentation has been modified accordingly, like docstring or example tutorials.

codecov · 2021-12-08T10:35:25Z

Codecov Report

Merging #6727 (3251093) into dev-v2.20.0 (eb1903f) will increase coverage by 0.02%.
The diff coverage is 82.75%.

@@               Coverage Diff               @@
##           dev-v2.20.0    #6727      +/-   ##
===============================================
+ Coverage        62.04%   62.06%   +0.02%     
===============================================
  Files              322      323       +1     
  Lines            25717    25744      +27     
  Branches          4356     4363       +7     
===============================================
+ Hits             15955    15977      +22     
- Misses            8913     8918       +5     
  Partials           849      849

Flag	Coverage Δ
unittests	`62.03% <82.75%> (+0.02%)`	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
mmdet/apis/train.py	`15.66% <16.66%> (-1.01%)`	⬇️
mmdet/utils/__init__.py	`100.00% <100.00%> (ø)`
mmdet/utils/misc.py	`100.00% <100.00%> (ø)`

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update eb1903f...3251093. Read the comment docs.

mmdet/utils/misc.py

tools/train.py

mmdet/apis/train.py

mmdet/utils/misc.py

tests/test_utils/test_misc.py

mmdet/apis/train.py

…dev-resume

* pre-commit * pre-commit * pre-commit * use cfg.get() to get auto_resume in apis/train.py

…b#6727) * pre-commit * pre-commit * pre-commit * use cfg.get() to get auto_resume in apis/train.py

pre-commit

8714079

Czm369 requested a review from hhaAndroid December 8, 2021 10:43

hhaAndroid reviewed Dec 9, 2021

View reviewed changes

mmdet/utils/misc.py Outdated Show resolved Hide resolved

pre-commit

edaa3d5