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.