-
Notifications
You must be signed in to change notification settings - Fork 2.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Enhance] New-style CPU training and inference. #1251
Conversation
Codecov Report
@@ Coverage Diff @@
## master #1251 +/- ##
==========================================
- Coverage 90.29% 90.26% -0.03%
==========================================
Files 130 130
Lines 7549 7553 +4
Branches 1256 1257 +1
==========================================
+ Hits 6816 6818 +2
- Misses 526 528 +2
Partials 207 207
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
The file conflict would be raised after #1247 merged. Wait for resolve... |
* [Enhance] New-style CPU training and inference. * assert mmcv version * SyncBN to BN in training and testing * SyncBN to BN in training and testing * upload untracked files to this branch * delete gpu_ids * fix bugs * assert args.gpu_id in train.py * use cfg.gpu_ids = [args.gpu_id] * use cfg.gpu_ids = [args.gpu_id] * fix typo * fix typo * fix typos
Motivation
Refers to open-mmlab/mmdetection#7016, and it's based on open-mmlab/mmcv#1621
Also solute to open-mmlab/mmocr#752.
Note: Due to modification of
mmcv/parallel/_functions.py
, see pr here.Due to deletion of above lines, so if use CPU function in this pr but mmcv is not the latest version, the error would be raised:
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 3, 3], but got 5-dimensional input of size [1, 4, 3, 128, 128] instead