• YosuaMichael's avatar
    Adding the huge vision transformer from SWAG (#5721) · 63576c9f
    YosuaMichael authored
    
    
    * Add vit_b_16_swag
    
    * Better handling idiom for image_size, edit test_extended_model to handle case where number of param differ from default due to different image size input
    
    * Update the accuracy to the experiment result on torchvision model
    
    * Fix typo missing underscore
    
    * raise exception instead of torch._assert, add back publication year (accidentally deleted)
    
    * Add license information on meta and readme
    
    * Improve wording and fix typo for pretrained model license in readme
    
    * Add vit_l_16 weight
    
    * Update README.rst
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Update the accuracy meta on vit_l_16_swag model to result from our experiment
    
    * Add vit_h_14_swag model
    
    * Add accuracy from experiments
    
    * Add to vit_h_16 model to hubconf.py
    
    * Add docs and expected pkl file for test
    
    * Remove legacy compatibility for ViT_H_14 model
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    
    * Test vit_h_14 with smaller image_size to speedup the test
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    63576c9f
test_models.py 31.2 KB