"examples/tensorflow/dgi/gcn.py" did not exist on "0a78dbe12a90845e1010adcf76f71f04f7386bd1"
- 28 Apr, 2022 3 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: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 10 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>
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
* Document MobileNetV3 quantization approach * Change the URL.
-
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 2 commits
-
-
Vasilis Vryniotis authored
* 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
-
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
-
- 21 Apr, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Removing `task`, `architecture` and `quantization` * Fix mypy * Remove size field * Remove unused import. * Fix mypy * Remove size from schema list. * update todo * Simplify with assert * Adding min_size to all models. * Update RAFT min size to 128
-
YosuaMichael authored
* Remove publication_year and interpolation meta * Add type to _COMMON_META and _COMMON_SWAG_META to prevent error from mypy check * Remove test to check interpolation and publication_year meta Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 20 Apr, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Convert weights only if `old_key` is in `state_dict` * Fix linter
-
- 19 Apr, 2022 4 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
-
tcmyxc authored
the assert msg should be same Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Shijie authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 18 Apr, 2022 1 commit
-
-
Kazuki Adachi authored
-
- 11 Apr, 2022 1 commit
-
-
YosuaMichael authored
* Add SWAG model that only the linear classifier head is finetuned with frozen trunk weight * Add accuracy from experiments * Change name from SWAG_LC to SWAG_LINEAR * Add comment on SWAG_LINEAR weight * Remove the comment docs (moved to PR description), and add the PR url as recipe. Also change name of previous swag model to SWAG_E2E_V1
-
- 06 Apr, 2022 3 commits
-
-
Vasilis Vryniotis authored
* Add MaskRCNN improved weights * Adding recipe URL
-
Vasilis Vryniotis authored
* Add FasterRCNN improved weights * Add recipe URL * Update publication_year field
-
Vasilis Vryniotis authored
* Add RetinaNet improved weights * Add weights. * Change publication date.
-
- 05 Apr, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Use frozen BN only if pre-trained. * Add LSJ and ability to from scratch training. * Fixing formatter * Adding `--opt` and `--norm-weight-decay` support in Detection. * Fix error message * Make ScaleJitter proportional. * Adding more norm layers in split_normalization_params. * Add FixedSizeCrop * Temporary fix for fill values on PIL * Fix the bug on fill. * Add RandomShortestSize. * Skip resize when an augmentation method is used. * multiscale in [480, 800] * Add missing star * Add new RetinaNet variant. * Add tests. * Update expected file for old retina * Fixing tests * Add FrozenBN to retinav2 * Fix network initialization issues * Adding BN support in MaskRCNNHeads and FPN * Adding support of FasterRCNNHeads * Introduce norm_layers in backbone utils. * Bigger RPN head + 2x rcnn v2 models. * Adding gIoU support to retinanet * Fix assert * Add back nesterov momentum * Rename and extend `FastRCNNConvFCHead` to support arbitrary FCs * Fix linter
-
YosuaMichael authored
* Add vit_b_16_swag * Better handling idiom for image_size, edit test_extended_model to handle case where number of param differ from default due to different image size input * Update the accuracy to the experiment result on torchvision model * Fix typo missing underscore * raise exception instead of torch._assert, add back publication year (accidentally deleted) * Add license information on meta and readme * Improve wording and fix typo for pretrained model license in readme * Add vit_l_16 weight * Update README.rst Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update the accuracy meta on vit_l_16_swag model to result from our experiment * Add vit_h_14_swag model * Add accuracy from experiments * Add to vit_h_16 model to hubconf.py * Add docs and expected pkl file for test * Remove legacy compatibility for ViT_H_14 model Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Test vit_h_14 with smaller image_size to speedup the test Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 04 Apr, 2022 1 commit
-
-
YosuaMichael authored
* Add regnet_y_128gh_swag weight * Add default weight for regnet_y_128gf * Add the accuracy from experiments
-
- 01 Apr, 2022 3 commits
-
-
YosuaMichael authored
* Add regnet_y_16gf and regnet_y_32gf from swag * Format with ufmt * Add the experiment accuracy Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
YosuaMichael authored
* Add vit_b_16_swag * Better handling idiom for image_size, edit test_extended_model to handle case where number of param differ from default due to different image size input * Update the accuracy to the experiment result on torchvision model * Fix typo missing underscore * raise exception instead of torch._assert, add back publication year (accidentally deleted) * Add license information on meta and readme * Improve wording and fix typo for pretrained model license in readme * Add vit_l_16 weight * Update README.rst Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update the accuracy meta on vit_l_16_swag model to result from our experiment Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Vasilis Vryniotis authored
* Detection recipe enhancements * Add back nesterov momentum
-