GitLab

* Indicate strings can be used to specify

* quotes

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

* Adding MViT v2 architecture (#6105)

* Adding mvitv2 architecture

* Fixing memory issues on tests and minor refactorings.

* Adding input validation

* Adding docs and minor refactoring

* Add `min_temporal_size` in the supported meta-data.

* Switch Tuple[int, int, int] with List[int] to support easier the 2D case

* Adding more docs and references

* Change naming conventions of classes to follow the same pattern as MobileNetV3

* Fix test breakage.

* Update todos

* Performance optimizations.

* Add support to MViT v1 (#6179)

* Switch implementation to v1 variant.

* Fix docs

* Adding back a v2 pseudovariant

* Changing the way the network are configured.

* Temporarily removing v2

* Adding weights.

* Expand _squeeze/_unsqueeze to support arbitrary dims.

* Update references script.

* Fix tests.

* Fixing frames and preprocessing.

* Fix std/mean values in transforms.

* Add permanent Dropout and update the weights.

* Update accuracies.

* Fix documentation

* Remove unnecessary expected file.

* Skip big model test

* Rewrite the configuration logic to reduce LOC.

* Fix mypy

* Add new .. betastatus:: directive to document Beta APIs

* Also add it for the fine-grained video API

* Add directive for all builders and pages of Detection module

* Also segmentation and video models

* Docs: build with Sphinx 5

* Sphinx 5 no longer accepts language = None
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Remove `(N, T, H, W, C) => (N, T, C, H, W)` conversion on presets

* Update docs.

* Fix the tests

* Use `output_format` for `read_video()`

* Use `output_format` for `Kinetics()`

* Adding input descriptions on presets

* To resolve issue #5964

Add note for resnet architecture

* Update resnet.py

* Update resnet.py

* Update resnet.rst

* Fix stylings

* Add the same notes on model builders

* Improve description

* Apply the change everywhere

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

* Adding an MLP block.

* Adding documentation

* Update typos.

* Fix inplace for Dropout.

* Apply recommendations from code review.

* Making changes on pre-trained models.

* Fix linter

* add swin_s and swin_b variants

* fix swin_b params

* fix n parameters and acc numbers

* adding missing acc numbers

* apply ufmt

* Updating `_docs` to reflect training recipe

* Fix exted for swin_b
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Remove models.rst

* Remove '- New'

* Put back torchhub section where it originally was

* Doc revamp for optical flow models

* Some more

* Classif models

* Detection

* Segmentation

* quantization

* Video

* optical flow

* tests

* Fix docs

* Fix Video dataset

* Consistency for RAFT dataset names

* use ImageNet-1K

* Use COCO-val2017-VOC-labels for segmentation

* formatting

* Adding docs for quantized models.

* Adding docs for video models.

* Adding docs for segmentation models.

* Adding docs for optical flow models.

* Adding docs for detection models.

* Fix typo.

* Make changes from code-review.

* try re arranging docs

* rewrite docs a bit

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

* Improving the auto-gen doc.

* Adding details for AlexNet, ConvNext, DenseNet, EfficientNets, GoogLeNet and InceptionV3.

* Fixing location of `_docs`

* Adding docs in the remaining classification models.

* Fix linter

* docs: Added quantized ResNeXt to new docs

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

* Minor updates on model examples.

* Improving wording of auto-generated docs.

* Add general info for pre-trained weights.

* Updating torch hub

* Minor updates

* Make lengthy meta-data partially visible

* Adding meta-data and reference info.

* Minor corrections

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

* Moving Torch hub section at the end
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* added note

* quantize = True higlighted in the note.

* Keep "Large" in docstring
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Create resnet_quant.rst

* add resnet quant

* refactor docs

* Minor fix

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

* [DOC] Add quantized inception updated documentation.

* Add missing file.

* Apply suggestions from code review

[ENH] Improve doc thanks to various code review comments.
Co-authored-by: Aditya Oke <47158509+oke-aditya@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>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Adding code examples for image classification + quant

* Adding code example detection

* Adding code example segmentation

* Adding code example for video classification

* Adding information on how to use the new API.

* Putting back the comma.

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

* Remove output to avoid staleness from flakiness.

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

* Adding `__repr__` in presets

* Adds `describe()` methods to all presets.

* Adding transform descriptions in the documentation.

* Change "preprocessing" to "inference"

* Remove old "minimum input size" from docstrings.

* Remove "currently only XYZ weights available"

* Fix description of wide_resnet101_2

* Make display license URLs as links.

* Clarify the order of dims of min_size.

* Remove lengthy keypoint_names from meta-table.

* added quantized mobilenetv2 docs

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

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>

* Add Docs for QuantizedGoogleNet

* ufmt

* Use explicit _QuantizedWeights or _Weights suffix

* [FEAT] Add distance IoU and distance IoU loss + some tests (WIP for tests).

* [FIX] Remove URL from docstring + remove assert since it causes a big performance drop.

* [FIX] eps isn't None.

* [TEST] Update existing box dIoU test + add dIoU loss tests (inspired from cIoU ones).

* [ENH] Some pre-commit fixes + remove print + mypy.

* [ENH] Pass the device in the assertion for the dIoU loss test.

* [FIX] Remove type hints from the dIoU box test.

* [ENH] Refactor box and loss for dIoU functions + fix half tests.

* [FIX] Precommits fix.

* [ENH] Some improvement for the distance IoU tests thanks to code review.

* [ENH] Upcast in distance boxes computation to avoid overflow.

* [ENH] Revert the refactor of distance IoU loss back since it introduced a bug and can be slow.

* Precommit fix.

* [FIX] Few changes introduced by merge conflict.

* Add code reference

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

* 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

* Add mask rcnn to instance segmentation

* Add docs a use nicolas suggestion

* remov param

* remov docsstring, edit docs

* Add one section level
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* 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>