GitLab

* Reinstate and deprecate `model_urls` and `quant_model_urls`

* Apply suggestions from code review
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Move todo location

* Add alexnet

* Add densenet

* Add efficientnet

* Add googlenet.

* Add inception.

* Add mobilenetv3

* Add regnet

* Add resnet

* Add shufflenetv2

* Fix linter

* Add squeezenet

* Add vgg

* Add vit

* Add quantized googlenet

* Add quantized inceptionv3

* Add quantized mobilenet_v3

* Add quantized resnet

* Add quantized shufflenetv2

* Fix incorrect imports

* Add faster_rcnn

* Add fcos

* Add keypoint rcnn

* Add mask rcnn

* Add retinanet

* Add ssd

* Add ssdlite

* Add deeplabv3

* Add fcn

* Add lraspp.

* Add video resnet

* Removing weights for shufflenetv2_x1.5 and shufflenetv2_x2.0

* Update the comments
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* POC

* Update torchvision/models/resnet.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Apply suggestions from code review
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Fix tests

* ufmt

* Remove useless docstring
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Adding ciou and diou support in `_box_loss()`

* Fix linter

* Addressing comments for nits

* Add Docs for QuantizedGoogleNet

* ufmt

* Use explicit _QuantizedWeights or _Weights suffix

* Fixing AlexNet references

* Fix doc rendering

* 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

* Document Keypoint RCNN separately

* Move Keypoint detection into its own section

* ufmt

* Remove references to non-existing weights

* Update torchvision/models/mnasnet.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/mnasnet.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* apply ufmt
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* docs: Added SSD300 to the new doc

To resolve #5897

* Minor fixes

* ufmt
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* updating mnasnet docs

* fix bug

* fix lint

* fix end of file

* fix linting

* Add missing weights
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update ShuffleNetV2 docstring

* Add entry for ShuffleNet V2 in new doc

* Minor fixes
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* init

* init

* Update torchvision/models/detection/faster_rcnn.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Update torchvision/models/detection/faster_rcnn.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Add inception

* Fix googlenet bug

* Fix bug add file

* Update torchvision/models/inception.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* add docs

* Small fix

* Update docs/source/models/lraspp.rst
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* docs: Added FCN to new doc

* docs: Updated docstring

* docs: Fixed docstring

* Update docs/source/models/fcn.rst
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* Apply suggestions from code review
Co-authored-by: FG Fernandez <–26927750+frgfm@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* docs: Added SSDLite to new doc

* docs: Updated docstring

* docs: Fixed docstring

* Apply suggestions from code review
Co-authored-by: FG Fernandez <–26927750+frgfm@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Add shufflenetv2 1.5 and 2.0 weights

* Update recipe

* Add to docs

* Use resize_size=232 for eval and update the result

* Add quantized shufflenetv2 large

* Update docs and readme

* Format with ufmt

* Add to hubconf.py

* Update readme for classification reference

* Fix reference classification readme

* Fix typo on readme

* Update reference/classification/readme

* Start adding

* Finish adding docs

* Fix other bugs

* Update torchvision/models/mobilenetv3.py

* Update torchvision/models/mobilenetv3.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* init

* minor change

* typo
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* refactor doc

* refactor code

* add fcos.rst

* add fcos
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* docs: Added MaskRCNN to new doc

* docs: Updated docstring

* ufmt
Co-authored-by: FG Fernandez <–26927750+frgfm@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Add revamped docs for video classification models

* EOL

* docs: Added entry for MobileNet V2 in new doc

* docs: Updated MobileNetV2 docstring

* docs: Fixed docstring
Co-authored-by: FG Fernandez <–26927750+frgfm@users.noreply.github.com>

* add swin transformer

* Update swin_transformer.py

* Update swin_transformer.py

* fix lint

* fix lint

* refactor code

* add swin_transformer

* Update swin_transformer.py

* fix bug

* refactor code

* fix lint

* update init_weights

* move shift_window into attention

* refactor code

* fix bug

* Update swin_transformer.py

* Update swin_transformer.py

* fix lint

* add patch_merge

* fix bug

* Update swin_transformer.py

* Update swin_transformer.py

* Update swin_transformer.py

* refactor code

* Update swin_transformer.py

* refactor code

* fix lint

* refactor code

* add swin_tiny

* add swin_tiny.pkl

* fix lint

* Delete ModelTester.test_swin_tiny_expect.pkl

* add swin_tiny

* add

* add Optional to bias

* update init weights

* update init_weights and add no weight decay

* add no weight decay

* add set_weight_decay

* add set_weight_decay

* fix lint

* fix lint

* add lr_cos_min

* add other swin models

* Update torchvision/models/swin_transformer.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* refactor doc

* Update utils.py

* Update train.py

* Update train.py

* Update swin_transformer.py

* update model builder

* fix lint

* add

* Update torchvision/models/swin_transformer.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Update torchvision/models/swin_transformer.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* update other model

* simplify the model name just like ViT

* add lr_cos_min

* fix lint

* fix lint

* Update swin_transformer.py

* Update swin_transformer.py

* Update swin_transformer.py

* Delete ModelTester.test_swin_tiny_expect.pkl

* add swin_t

* refactor code

* Update train.py

* add swin_s

* ignore a error of mypy

* Update swin_transformer.py

* fix lint

* add swin_b

* add swin_l

* refactor code

* Update train.py

* move relative_position_bias to __init__

* fix formatting

* Revert "fix formatting"

This reverts commit 41faba232668f7ac4273a0cf632c0d0130c7ce9c.

* Revert "move relative_position_bias to __init__"

This reverts commit f0615440bf18617dc0e5dc4839bd5ed27e5ed010.

* refactor code

* Remove deprecated meta-data from `_COMMON_META`

* fix linter

* add pretrained weights for swin_t

* fix format

* apply ufmt

* add documentation

* update references README

* adding new style docs

* update pre-trained weights values

* remove other variants

* fix typo

* Remove expect for the variants not yet supported
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Joao Gomes <jdsgomes@fb.com>

* Document MobileNetV3 quantization approach

* Change the URL.

* [DOC] Add alexnet new documentation.

* [FIX] Fix wrong alexnet header.

* [FIX] Fix the AlexNet paper link.
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Start doc revamp for detection models

* Minor cleanup

* Use list of tuples for metrics

* added dense net docs

* fix models_new.rst

* initial commit

* formatting

* minor change

* replaced pdf link with webpage link

* replaced pdf link with webpage link

* minor

* minor
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* add convnext.rst

* refactor code

* add convnext
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* [FEAT] Add efficientnetv2 doc.

* [FIX] Precommits fix.

* Apply suggestions from code review
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* [DOC] Add new EfficientNet doc.

* [DOC] Add missing file.

* [FIX] Precommits fix.

* [FIX] Precommits fix.

* [ENH] Integrate few documentation comments from code review

* Update docs/source/models/efficientnet.rst
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* 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