Would features_only model work as backbone for torchvision model #602
Answered
by
rwightman
AnhMinhTran
asked this question in
Q&A
-
Anyone here has tried putting using the feature extractor arg to make a backbone for torch.vision model and did it work for you guys? |
Beta Was this translation helpful? Give feedback.
Answered by
rwightman
May 1, 2021
Replies: 1 comment 1 reply
-
@AnhMinhTran no, it would not work via a factory method... if you created a |
Beta Was this translation helpful? Give feedback.
1 reply
Answer selected by
AnhMinhTran
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@AnhMinhTran no, it would not work via a factory method... if you created a
.feature_info
(list of dicts) like the models here that marks the module names for each feature level, you could wrap it in one of my feature extractors. Not sure it's worth the effort though, most models that are there are also here. And often have newer / better trained weights here.