Skip to content

Would features_only model work as backbone for torchvision model #602

Answered by rwightman
AnhMinhTran asked this question in Q&A
Discussion options

You must be logged in to vote

@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.

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@AnhMinhTran
Comment options

Answer selected by AnhMinhTran
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants