- 18 May, 2022 3 commits
-
-
Nicolas Hug authored
* 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
-
Aditya Oke authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Vasilis Vryniotis authored
* 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.
-
- 17 May, 2022 4 commits
-
-
Vasilis Vryniotis authored
* 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
-
WuZhe authored
* update paper link for FCOS refence * remove 'updated version' Co-authored-by: -
F-G Fernandez authored
* docs: Added quantized ResNeXt to new docs * docs: Fixed docstring Co-authored-by: -
F-G Fernandez authored
-
- 16 May, 2022 7 commits
-
-
Zhiqiang Wang authored
-
YosuaMichael authored
* Add weight for mnasnet0_75 and mnasnet1_3 * Fix missing comma * Add PR url as recipe, and update the metrics * Add weights to legacy handler * Update docs to specify there are weights available
-
Vasilis Vryniotis authored
* Prefixing `_get_enum_from_fn` with underscore * Exposing `get_weight` to Torch Hub
-
Nicolas Hug authored
-
Vasilis Vryniotis authored
-
Abhijit Deo authored
-
Abhijit Deo authored
* added note * quantize = True higlighted in the note. * Keep "Large" in docstring Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
- 13 May, 2022 2 commits
-
-
Hu Ye authored
* Create resnet_quant.rst * add resnet quant * refactor docs * Minor fix * Nit Co-authored-by: -
Yassine Alouini authored
* [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:
-
- 12 May, 2022 4 commits
-
-
Vasilis Vryniotis authored
* 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.
-
Lezwon Castelino authored
* added quantized mobilenetv2 docs * remove quotes Co-authored-by: -
Nicolas Hug authored
-
Vasilis Vryniotis authored
* Reinstate and deprecate `model_urls` and `quant_model_urls` * Apply suggestions from code review Co-authored-by:
-
- 11 May, 2022 2 commits
-
-
Nicolas Hug authored
* POC * Update torchvision/models/resnet.py Co-authored-by:
-
Vasilis Vryniotis authored
* Adding ciou and diou support in `_box_loss()` * Fix linter * Addressing comments for nits
-
- 10 May, 2022 1 commit
-
-
Nicolas Hug authored
* Add Docs for QuantizedGoogleNet * ufmt * Use explicit _QuantizedWeights or _Weights suffix
-
- 09 May, 2022 4 commits
-
-
Nicolas Hug authored
* Fixing AlexNet references * Fix doc rendering
-
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
-
Nicolas Hug authored
-
Philip Meier authored
-
- 04 May, 2022 3 commits
-
-
Nicolas Hug authored
* Document Keypoint RCNN separately * Move Keypoint detection into its own section * ufmt
-
Nicolas Hug authored
-
Joao Gomes authored
* Remove references to non-existing weights * Update torchvision/models/mnasnet.py Co-authored-by:
-
- 03 May, 2022 8 commits
-
-
puhuk authored
* docs: Added SSD300 to the new doc To resolve #5897 * Minor fixes * ufmt Co-authored-by: -
Joao Gomes authored
* updating mnasnet docs * fix bug * fix lint * fix end of file * fix linting * Add missing weights Co-authored-by: -
Zhiqiang Wang authored
* Update ShuffleNetV2 docstring * Add entry for ShuffleNet V2 in new doc * Minor fixes Co-authored-by: -
Abhijit Deo authored
* init * init * Update torchvision/models/detection/faster_rcnn.py Co-authored-by:
-
Aditya Oke authored
* Add inception * Fix googlenet bug * Fix bug add file * Update torchvision/models/inception.py Co-authored-by: -
Aditya Oke authored
* add docs * Small fix * Update docs/source/models/lraspp.rst Co-authored-by: -
F-G Fernandez authored
* docs: Added FCN to new doc * docs: Updated docstring * docs: Fixed docstring * Update docs/source/models/fcn.rst Co-authored-by:
FG Fernandez <26927750+frgfm@users.noreply.github.com> Co-authored-by:
-
F-G Fernandez authored
* docs: Added SSDLite to new doc * docs: Updated docstring * docs: Fixed docstring * Apply suggestions from code review Co-authored-by:
-
- 28 Apr, 2022 2 commits
-
-
YosuaMichael authored
* 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
-
Aditya Oke authored
* Start adding * Finish adding docs * Fix other bugs * Update torchvision/models/mobilenetv3.py * Update torchvision/models/mobilenetv3.py Co-authored-by:
-
