1. 09 Dec, 2021 2 commits
  2. 08 Dec, 2021 11 commits
  3. 07 Dec, 2021 5 commits
  4. 06 Dec, 2021 2 commits
  5. 05 Dec, 2021 2 commits
  6. 03 Dec, 2021 1 commit
  7. 02 Dec, 2021 3 commits
  8. 01 Dec, 2021 3 commits
  9. 30 Nov, 2021 5 commits
  10. 29 Nov, 2021 5 commits
  11. 27 Nov, 2021 1 commit
    • Yiwen Song's avatar
      Adding ViT to torchvision/models (#4594) · 47281bbf
      Yiwen Song authored
      
      
      * [vit] Adding ViT to torchvision/models
      
      * adding pre-logits layer + resolving comments
      
      * Fix the model attribute bug
      
      * Change version to arch
      
      * fix failing unittests
      
      * remove useless prints
      
      * reduce input size to fix unittests
      
      * Increase windows-cpu executor to 2xlarge
      
      * Use `batch_first=True` and remove classifier
      
      * Change resource_class back to xlarge
      
      * Remove vit_h_14
      
      * Remove vit_h_14 from __all__
      
      * Move vision_transformer.py into prototype
      
      * Fix formatting issue
      
      * remove arch in builder
      
      * Fix type err in model builder
      
      * address comments and trigger unittests
      
      * remove the prototype import in torchvision.models
      
      * Adding vit back to models to trigger CircleCI test
      
      * fix test_jit_forward_backward
      
      * Move all to prototype.
      
      * Adopt new helper methods and fix prototype tests.
      
      * Remove unused import.
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      Co-authored-by: default avatarVasilis Vryniotis <vvryniotis@fb.com>
      47281bbf