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

other models? #20

Closed
EdenBelouadah opened this issue Feb 2, 2022 · 4 comments
Closed

other models? #20

EdenBelouadah opened this issue Feb 2, 2022 · 4 comments

Comments

@EdenBelouadah
Copy link

Hello
Firstly, thank you for this great work.
Please, is it possible to use the code with whatever teacher/student model from mmdetection?
I am interested to distill knowledge from a large model to a very light student such as MobileNet, and use it for fast inference.

Thank you in advance

@HikariTJU
Copy link
Owner

If you mean backbone, then yes.
You can edit the backbone option in the config file to navigate different teacher/student.

@EdenBelouadah
Copy link
Author

Thank you for your answer.
Yes, I would like to have a MobileNet as a student. Can you please tell me what to modify in this config file:

https://github.com/HikariTJU/LD/blob/main/configs/ld/ld_r50_gflv1_r101_fpn_coco_1x.py

to have a resnet 50 as a teacher and a mobilenet as a student?

I appreciate your help

@HikariTJU
Copy link
Owner

HikariTJU commented Feb 3, 2022

Student: https://github.com/HikariTJU/LD/blob/main/docs/tutorials/customize_models.md
Teacher: insert

    teacher_config='configs/gfl/gfl_r50_fpn_1x_coco.py',
    teacher_ckpt=PATH_TO_YOUR_TEACHER_CKPT,

into model config

@EdenBelouadah
Copy link
Author

Thank you very much. This is so helpful.

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

No branches or pull requests

2 participants