Skip to content
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

[Feature] Support Resnet strikes back #1390

Merged
merged 5 commits into from
Apr 5, 2022
Merged

Conversation

MengzhangLI
Copy link
Contributor

@MengzhangLI MengzhangLI commented Mar 17, 2022

Motivation

Transfer learning of paper 'Resnet strikes back: An improved training procedure in timm'.

paper URL https://arxiv.org/pdf/2110.00476

Related PR: open-mmlab/mmdetection#7001.

Notification:

(1) Current MMSegmentation default ResNet backbone is ResNetV1c, which performs better than standard ResNet backbone.
(2) This PR only implements on PSPNet.
(3) With more RSB (Resnet strikes back) experiments on different decoder heads (like DeepLabV3, FCN and so on) and RSB strategy on ResNetV1c backbone, we would consider open an independent resnet_strikes_back folder in ./configs/ like MMDetection.

Results

Method Backbone optimizer learning rate weight decay Detail config Dataset mIoU mIoU (ms + flip)
PSPNet ResNet RSB R50-D8 AdamW 0.0005 0.05 KNET Cityscapes 78.47 79.45
PSPNet ResNet V1c R50-D8 SGD 0.01 0.0005 None Cityscapes 78.55 79.49
PSPNet ResNet R50-D8 SGD 0.01 0.0005 None Cityscapes 78.22 79.46
                 
PSPNet ResNet RSB R50-D32 AdamW 0.0005 0.05 KNET Cityscapes 74.09 77.18
PSPNet ResNet V1c R50-D32 SGD 0.01 0.0005 None Cityscapes 73.88 76.85
PSPNet ResNet R50-D32 SGD 0.01 0.0005 None Cityscapes 72.61 75.51

Where KNET means learning policy referred from K-Net semantic segmentation configs, which works well with AdamW optimizer:

optimizer_config = dict(grad_clip=dict(max_norm=1, norm_type=2))
# learning policy
lr_config = dict(
    _delete_=True,
    policy='step',
    warmup='linear',
    warmup_iters=1000,
    warmup_ratio=0.001,
    step=[60000, 72000],
    by_epoch=False)

@MengzhangLI MengzhangLI self-assigned this Mar 17, 2022
@codecov
Copy link

codecov bot commented Mar 17, 2022

Codecov Report

Merging #1390 (4f02398) into master (3d0c2eb) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master    #1390   +/-   ##
=======================================
  Coverage   90.39%   90.39%           
=======================================
  Files         133      133           
  Lines        7906     7906           
  Branches     1318     1318           
=======================================
  Hits         7147     7147           
  Misses        536      536           
  Partials      223      223           
Flag Coverage Δ
unittests 90.39% <ø> (ø)

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

Impacted Files Coverage Δ
mmseg/core/evaluation/class_names.py 88.33% <ø> (ø)

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 3d0c2eb...4f02398. Read the comment docs.

* Add training startup documentation

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix
configs/pspnet/README.md Outdated Show resolved Hide resolved
@MeowZheng MeowZheng merged commit c9f507e into open-mmlab:master Apr 5, 2022
mob5566 pushed a commit to mob5566/mmsegmentation that referenced this pull request Apr 13, 2022
* [Feature] Support Resnet strikes back

* fix url

* [Feature] Add multi machine `dist_train`. (open-mmlab#1383)

* Add training startup documentation

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* modify R-50b rsb

Co-authored-by: FangjianLin <93248678+linfangjian01@users.noreply.github.com>
ZhimingNJ pushed a commit to AetrexTechnology/mmsegmentation that referenced this pull request Jun 29, 2022
* [Feature] Support Resnet strikes back

* fix url

* [Feature] Add multi machine `dist_train`. (open-mmlab#1383)

* Add training startup documentation

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* modify R-50b rsb

Co-authored-by: FangjianLin <93248678+linfangjian01@users.noreply.github.com>
@MengzhangLI MengzhangLI deleted the rsb branch July 15, 2022 03:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants