• NielsRogge's avatar
    Add BeitForSemanticSegmentation (#14096) · e20faa6f
    NielsRogge authored
    
    
    * Add first draft
    
    * Make forward pass work
    
    * Improve conversion script
    
    * Add notebook that checks if it works
    
    * Add BeitForSemanticSegmentation to the tests
    
    * More improvements
    
    * Make BeitForSemanticSegmentation consistent with Segformer
    
    * Small bug fix
    
    * Add BeitForSemanticSegmentation to docs
    
    * Make sure model doesn't output hidden states when the user doesn't want to
    
    * Make it possible to convert the large model
    
    * Fix issue
    
    * Fix conversion script for large model
    
    * Add auxiliary_head option to semantic segmentation model
    
    * Apply suggestions from @sgugger's review
    
    * Apply suggestions from code review
    
    * Fix failing test
    Co-authored-by: default avatarLysandre <lysandre.debut@reseau.eseo.fr>
    e20faa6f
check_repo.py 24.1 KB