• NielsRogge's avatar
    Add DeiT (PyTorch) (#11056) · 9f126097
    NielsRogge authored
    * First draft of deit
    
    * More improvements
    
    * Remove DeiTTokenizerFast from init
    
    * Conversion script works
    
    * Add DeiT to ViT conversion script
    
    * Add tests, add head model, add support for deit in vit conversion script
    
    * Update model checkpoint names
    
    * Update image_mean and image_std, set resample to bicubic
    
    * Improve docs
    
    * Docs improvements
    
    * Add DeiTForImageClassificationWithTeacher to init
    
    * Address comments by @sgugger
    
    * Improve feature extractors
    
    * Make fix-copies
    
    * Minor fixes
    
    * Address comments by @patil-suraj
    
    * All models uploaded
    
    * Fix tests
    
    * Remove labels argument from DeiTForImageClassificationWithTeacher
    
    * Fix-copies, style and quality
    
    * Fix tests
    
    * Fix typo
    
    * Multiple docs improvements
    
    * More docs fixes
    9f126097
README.md 33.8 KB