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.

Sign up for GitHub

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

Jump to bottom

[Feature] Add RTMDet distillation cfg #544

Merged
merged 23 commits into from
Mar 1, 2023
Merged

[Feature] Add RTMDet distillation cfg #544

merged 23 commits into from
Mar 1, 2023

Conversation

HIT-cwh
Copy link
Contributor

@HIT-cwh HIT-cwh commented Feb 10, 2023 •
edited
Loading

Motivation

Support distilling rtmdet with mmrazor.

Modification

  1. PR#455 in mmrazor support specifying epochs to stop knowledge distillation.
  2. Add corresponding config files.
  3. Add a readme to introduce how to distill rtm detectors based on MMRazor.

Results

  1. rtm-s -> rtm-tiny Origin: 41.0 After distillation: 41.8.
  2. rtm-m -> rtm-s Origin: 44.6 After distillation: 45.7.
  3. rtm-l -> rtm-m Origin: 49.3 After distillation: 50.2.
  4. rtm-x -> rtm-l Origin: 51.4 After distillation: 52.3.

TODO

  1. Upload the checkpoint.

@CLAassistant
Copy link

CLAassistant commented Feb 10, 2023 •
edited
Loading

CLA assistant check
All committers have signed the CLA.

@RangeKing RangeKing changed the base branch from main to dev February 10, 2023 10:29
@RangeKing RangeKing changed the title [feature] Add rtm distillation cfg [Feature] Add RTMDet distillation cfg Feb 10, 2023
@HIT-cwh HIT-cwh changed the title [Feature] Add RTMDet distillation cfg [WIP] Add RTMDet distillation cfg Feb 13, 2023
pppppM
pppppM reviewed Feb 28, 2023
View reviewed changes
Copy link

@pppppM pppppM left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

  1. Add an installation guide for MMRazor
  2. Add docstring in configs

configs/rtmdet/distillation/README.md Outdated Show resolved Hide resolved
configs/rtmdet/distillation/README.md Outdated Show resolved Hide resolved
configs/rtmdet/distillation/README.md Outdated Show resolved Hide resolved
RangiLyu
RangiLyu reviewed Feb 28, 2023
View reviewed changes
configs/rtmdet/README.md Outdated Show resolved Hide resolved
@HIT-cwh @RangiLyu
Update configs/rtmdet/README.md
043a62d 
Co-authored-by: RangiLyu <lyuchqi@gmail.com>
@RangiLyu RangiLyu changed the title [WIP] Add RTMDet distillation cfg [Feature] Add RTMDet distillation cfg Feb 28, 2023
@hhaAndroid hhaAndroid merged commit 6f38b78 into open-mmlab:dev Mar 1, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@RangeKing RangeKing RangeKing left review comments

@pppppM pppppM pppppM left review comments

@RangiLyu RangiLyu RangiLyu approved these changes

@hhaAndroid hhaAndroid hhaAndroid approved these changes

Assignees
No one assigned
Labels
Feature:P0
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
6 participants
@HIT-cwh @CLAassistant @RangiLyu @hhaAndroid @RangeKing @pppppM