-
Notifications
You must be signed in to change notification settings - Fork 1.7k
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
Support to specify LR of DCN's conv_offset #344
Conversation
Codecov Report
@@ Coverage Diff @@
## master #344 +/- ##
==========================================
+ Coverage 64.76% 64.95% +0.18%
==========================================
Files 133 134 +1
Lines 7597 7632 +35
Branches 1336 1346 +10
==========================================
+ Hits 4920 4957 +37
+ Misses 2432 2425 -7
- Partials 245 250 +5
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
Task linked: CU-49u01p Support to specify LR of DCN's conv_offset |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Adding offset_lr_mult
to the init method may be confusing when users are not using the default optimizer constructor.
* Support to specify LR of DCN's conv_offset * Resolve comments & add unit test * Resolve formats * Fix CI for DCN * Mock DCN when cpu only * Use mock for cpu testing * Fix docstring and support ModulatedDCN * set offset_lr_mult as dcn's arguments, link CU-49u01p * fix lr bug * fall back to set LR in constructor * resolve comments
* tag 'v1.1.5': (81 commits) bump version to 1.1.5 (open-mmlab#603) Fix wrappers version comparison (open-mmlab#602) revert dcn (open-mmlab#597) lint markdown files (open-mmlab#592) bump version to 1.1.4 (open-mmlab#591) fix lint open-mmlab#580 (open-mmlab#590) Fixed fp16_optimizer state bug (open-mmlab#580) register deconv in CONV_LAYERS (open-mmlab#582) [enhance]: show grad norm in fp16 optimizer hook (open-mmlab#584) [Enhance] Switch to https://download.openmmlab.com (open-mmlab#583) Support to specify LR of DCN's conv_offset (open-mmlab#344) add torchvision roi_align with aligned=True (open-mmlab#581) Add runner builder (open-mmlab#570) [fix]: fix wrapper comparison of pytorch version (open-mmlab#572) Add missing by_epoch arg (open-mmlab#576) merge the calling of train/val_step and batch_processor into run_iter (open-mmlab#553) bump version to 1.1.3 (open-mmlab#569) fix deprecated wrappers exiting bug (open-mmlab#567) add switch for onnx exporter (open-mmlab#564) fix dcon forward and backward bug (open-mmlab#565) ...
This PR add an option
dcn_offset_lr_mult
inparams_cfg
. The modification enables to set the learning rate of DCN's conv offset layer to stabilize training.