• YosuaMichael's avatar
    Add SWAG Vision Transformer Weight (#5714) · 781b0f9c
    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
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    781b0f9c
test_extended_models.py 9.28 KB