Skip to content

GitLab

Switch branch/tag
  1. 09 May, 2022 1 commit
    • YosuaMichael's avatar
      Adding resnext101 64x4d model (#5935) · 4c02f103
      YosuaMichael authored 
      * Add resnext101_64x4d model definition

* Add test for resnext101_64x4d

* Add resnext101_64x4d weight

* Update checkpoint to use EMA weigth

* Add quantization model signature for resnext101_64x4d

* Fix class name and update accuracy using 1 gpu and batch_size=1

* Apply ufmt

* Update the quantized weight and accuracy that we still keep the training log

* Add quantized expect file

* Update docs and fix acc1

* Add recipe for quantized to PR

* Update models.rst
      4c02f103
  3. 05 May, 2022 1 commit
  5. 04 May, 2022 2 commits
  7. 03 May, 2022 8 commits
  9. 28 Apr, 2022 5 commits
  11. 27 Apr, 2022 8 commits
  13. 26 Apr, 2022 3 commits
  15. 25 Apr, 2022 5 commits
  17. 22 Apr, 2022 2 commits
    • Vasilis Vryniotis's avatar
      Restructuring metrics meta-data (#5859) · 6f016dd9
      Vasilis Vryniotis authored 
      * Restructuring metrics meta-data for detection, segmentation and optical flow.

* Renaming acc to pixel_acc for segmentation

* Restructure video meta-data.

* Restructure classification and quantization meta-data.

* Fix tests.

* Fix documentation
      6f016dd9
    • YosuaMichael's avatar
      Adding revamp docs for vision_transformers and regnet (#5856) · ca265374
      YosuaMichael authored 
      * Add docs for regnet, still need to update the comment docs on models

* Fix a little typo on .rst file

* Update regnet docstring

* Add vision_transformer docs, and fix typo on regnet docs

* Update docstring to make sure it does not exceed 120 chars per line

* Improve formatting

* Change the new line location for vision_transformer docstring
      ca265374
  19. 19 Apr, 2022 3 commits
    • Nicolas Hug's avatar
      Add docs for `SomeWeight.DEFAULT` fields (#5831) · e8cb0bac
      Nicolas Hug authored 
      * First PR for model doc revamp

* Deactivating fail on warning, temporarily

* Remove commnet

* Minor changes

* Typos

* Added TODO in Makefile

* Keep old models.rst file intact, move new docs into new models_new.rst file

* Added docs LOL
      e8cb0bac
    • Nicolas Hug's avatar
      Add docs for SqueezeNet (#5832) · f1cd8b23
      Nicolas Hug authored
      f1cd8b23
    • Nicolas Hug's avatar
      Model doc revamp - first PR (#5821) · 111fe85d
      Nicolas Hug authored 
      * First PR for model doc revamp

* Deactivating fail on warning, temporarily

* Remove commnet

* Minor changes

* Typos

* Added TODO in Makefile

* Keep old models.rst file intact, move new docs into new models_new.rst file
      111fe85d
  21. 05 Apr, 2022 1 commit
    • 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
  23. 23 Mar, 2022 1 commit