GitLab

* add test split for imagenet

* add infinite buffer size to shuffler

* disable weight download and state dict loading for model tests

* fix indent

* debug

* nuclear option

* revert unrelated change

* cleanup

* add explanation

* typo

* remove unused imports

* update lsun

* Adding multiweight support to Keypoint RCNN.

* Fix linter.

* Updated wide_resnet50_2 model weights

* Update Accs with batch-size 1.

* Update checkpoint and accs.
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

Co-authored-by: Francisco Massa <fvsmassa@gmail.com>

* Adding multiweight suport on Quant MobileNetV2 and MobileNetV3.

* Fixing enum name.

* Fixing lint.

* Clean up unnecessary quant builders and add quant weights for 0.5

* Fixing mypy.

* add prototype features

* add some JIT tests

* refactor input data handling

* refactor tests

* cleanup tests

* add BoundingBox feature

* mypy

* xfail torchscript tests for now

* cleanup

* fix imports

* add tests for builtin prototype datasets

* fix caltech101

* fix emnist

* fix mnist and variants

* add iopath as test requirement

* fix MNIST warning

* fix qmnist data generation

* fix cifar data generation

* add tests for imagenet

* cleanup

* Moving builder to the bottom to use proper typing.

* Renaming weights.

* Adding quantizated inception builder.

* Correct meta info.

* Fix linter.

* Removing init_weights to avoid exposing it on the class.

* modify processPacket to support fast seek

* add fastSeek to ProcessPacket decoder definition

* add fastseek flag to DecoderParametersStruct

* add fastseek flag to the process packet call

* no default params in C++ implementation

* enable flag in C++ implementation

* make order of parameters more normal

* register new seek with python api

* [somewhat broken] test suite for keyframes using pyav

* revert " changes

* add type annotations to init

* Adding tests

* linter

* Flake doesn't show up :|

* Change from unitest to pytest syntax

* add return type
Co-authored-by: Prabhat Roy <prabhatroy@fb.com>

* Reordering the builders to use proper typing.

* Adding additional meta-data on existing quantized models.

* Fixing meta on unquantized model.

* Adding quantized googlenet builder.

* undo inception move.

* Adding recipe information.

* Aligning exception with all other models.

* Adding prototype preprocessing on video references.

* Adding the rest of model builders on faster_rcnn.

torchvision compatible with torch 1.9.1 was missing in table of supported versions.

* Fix TORCH_HOME environment variable in docs

* Update docs with torch.hub

Updated the reference for loading a model to `torch.hub`.
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Adding prototype preprocessing on segmentation references.

* Adding prototype preprocessing on video references.

* Port semeion dataset

* Update torchvision/prototype/datasets/_builtin/semeion.py
Co-authored-by: Philip Meier <github.pmeier@posteo.de>

* explicitly convert the image array to torch.uint8

* explicitly convert the image array to torch.uint8
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* use `store_test_results` for prototype tests

* add failing test

* Revert "add failing test"

This reverts commit ebe2b37cb50461031a94131fe4bd3fae71b48e34.

* [references] Adding gradient clipping

* ufmt formatting

* remove apex code

* resolve naming issue

Fix for https://github.com/pytorch/vision/issues/4832

The unauthenticated git protocol on port 9418 is no longer supported.
Please see https://github.blog/2021-09-01-improving-git-protocol-security-github/ for more information.

* Update typing

* Fix bug

* Unblock mypy

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

* Adding multi-weight support to Quantized ResNet.

* Update references script to support testing quantized models with the new API.

* Handle quantized models correctly in ref script.

* Fixing references for quantization.

* Update training references from legacy models.

* Refactoring to share common parts.

* Expose misc ops at package level

* Adding documentation to the ops exposed

* Moving original builder at the bottom of the page to use proper typing.

* Adding multiweight support to inception.

* Update doc.