- 12 May, 2022 3 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 <contact@nicolas-hug.com>
-
Vasilis Vryniotis authored
-
- 10 May, 2022 1 commit
-
-
Nicolas Hug authored
* Add Docs for QuantizedGoogleNet * ufmt * Use explicit _QuantizedWeights or _Weights suffix
-
- 09 May, 2022 2 commits
-
-
Nicolas Hug authored
-
Nicolas Hug authored
* Fixing AlexNet references * Fix doc rendering
-
- 05 May, 2022 1 commit
-
-
Aditya Oke authored
* 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>
-
- 04 May, 2022 2 commits
-
-
Nicolas Hug authored
* Document Keypoint RCNN separately * Move Keypoint detection into its own section * ufmt
-
Joao Gomes authored
* 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>
-
- 03 May, 2022 8 commits
-
-
puhuk authored
* docs: Added SSD300 to the new doc To resolve #5897 * Minor fixes * ufmt Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Joao Gomes authored
* 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>
-
Zhiqiang Wang authored
* Update ShuffleNetV2 docstring * Add entry for ShuffleNet V2 in new doc * Minor fixes Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Abhijit Deo authored
* 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>
-
Aditya Oke authored
* Add inception * Fix googlenet bug * Fix bug add file * Update torchvision/models/inception.py Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Aditya Oke authored
* add docs * Small fix * Update docs/source/models/lraspp.rst Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
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:
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>
-
F-G Fernandez authored
* 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>
-
- 28 Apr, 2022 2 commits
-
-
Aditya Oke authored
* 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>
-
Abhijit Deo authored
* init * minor change * typo Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
- 27 Apr, 2022 8 commits
-
-
Nicolas Hug authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Hu Ye authored
* refactor doc * refactor code * add fcos.rst * add fcos Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
F-G Fernandez authored
* 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>
-
Nicolas Hug authored
* Add revamped docs for video classification models * EOL
-
F-G Fernandez authored
* 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>
-
Hu Ye authored
* 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>
-
Yassine Alouini authored
* [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>
-
Nicolas Hug authored
-
- 26 Apr, 2022 3 commits
-
-
Yassine Alouini authored
-
Nicolas Hug authored
* Start doc revamp for detection models * Minor cleanup * Use list of tuples for metrics
-
Lezwon Castelino authored
* added dense net docs * fix models_new.rst
-
- 25 Apr, 2022 4 commits
-
-
Abhijit Deo authored
* 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>
-
Hu Ye authored
* add convnext.rst * refactor code * add convnext Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Yassine Alouini authored
* [FEAT] Add efficientnetv2 doc. * [FIX] Precommits fix. * Apply suggestions from code review Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Yassine Alouini authored
* [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>
-
- 22 Apr, 2022 1 commit
-
-
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
-
- 19 Apr, 2022 2 commits
-
-
Nicolas Hug authored
-
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
-