Skip to content

GitLab

Switch branch/tag
  1. 08 Dec, 2021 1 commit
  3. 07 Dec, 2021 5 commits
  5. 06 Dec, 2021 2 commits
  7. 05 Dec, 2021 2 commits
  9. 03 Dec, 2021 1 commit
  11. 02 Dec, 2021 3 commits
  13. 01 Dec, 2021 3 commits
  15. 30 Nov, 2021 5 commits
  17. 29 Nov, 2021 5 commits
  19. 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
  21. 26 Nov, 2021 3 commits
    • Philip Meier's avatar
      Download model weights in parallel for prototype CI (#4772) · 29f38f17
      Philip Meier authored 
      * enable caching of model weights for prototype CI

* syntax

* syntax

* make cache dir dynamic

* increase verbosity

* fix

* use larget CI machine

* revert debug output

* [DEBUG] test env var usage in save_cache

* retry

* use checksum for caching

* remove env vars because expansion is not working

* syntax

* cleanup

* base caching on model-urls

* relax regex

* cleanup skips

* cleanup

* fix skipping logic

* improve step name

* benchmark without caching

* benchmark with external download

* debug

* fix manual download location

* debug again

* download weights in the background

* try parallel download

* add missing import

* use correct decoractor

* up resource_class

* fix wording

* enable stdout passthrough to see download during test

* remove linebreak

* move checkout up

* cleanup

* debug failing test

* temp fix

* fix

* cleanup

* fix regex

* remove explicit install of numpy
      29f38f17
    • Hu Ye's avatar
      support amp training for segmention models (#4994) · cca16993
      Hu Ye authored 
      

* support amp training for segmention models

* fix lint
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      cca16993
    • Philip Meier's avatar
      fix mobilenetv3 quantization state dict loading (#4997) · 58016b09
      Philip Meier authored
      58016b09
  23. 25 Nov, 2021 2 commits
    • Vasilis Vryniotis's avatar
      More Multiweight support cleanups (#4948) · 18cf5ab1
      Vasilis Vryniotis authored 
      * Updated the link for densenet recipe.

* Set default value of `num_classes` and `num_keypoints` to `None`

* Provide helper methods for parameter checks to reduce duplicate code.

* Throw errors on silent config overwrites from weight meta-data and legacy builders.

* Changing order of arguments + fixing mypy.

* Make the builders fully BC.

* Add "default" weights support that returns always the best weights.
      18cf5ab1
    • Yuchen Huang's avatar
      [iOS] Fix VisionTestApp setups (#4989) · 09e759ea
      Yuchen Huang authored
      09e759ea
  25. 24 Nov, 2021 1 commit
  27. 23 Nov, 2021 1 commit
  29. 22 Nov, 2021 5 commits