- 21 Apr, 2022 1 commit
-
-
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>
-
- 01 Apr, 2022 2 commits
-
-
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 authored
* Detection recipe enhancements * Add back nesterov momentum
-
- 22 Mar, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Moving basefiles outside of prototype and porting Alexnet, ConvNext, Densenet and EfficientNet. * Porting googlenet * Porting inception * Porting mnasnet * Porting mobilenetv2 * Porting mobilenetv3 * Porting regnet * Porting resnet * Porting shufflenetv2 * Porting squeezenet * Porting vgg * Porting vit * Fix docstrings * Fixing imports * Adding missing import * Fix mobilenet imports * Fix tests * Fix prototype tests * Exclude get_weight from models on test * Fix init files * Porting googlenet * Porting inception * porting mobilenetv2 * porting mobilenetv3 * porting resnet * porting shufflenetv2 * Fix test and linter * Fixing docs. * Porting Detection models (#5617) * fix inits * fix docs * Port faster_rcnn * Port fcos * Port keypoint_rcnn * Port mask_rcnn * Port retinanet * Port ssd * Port ssdlite * Fix linter * Fixing tests * Fixing tests * Fixing vgg test * Porting Optical Flow, Segmentation, Video models (#5619) * Porting raft * Porting video resnet * Porting deeplabv3 * Porting fcn and lraspp * Fixing the tests and linter * Porting docs, examples, tutorials and galleries (#5620) * Fix examples, tutorials and gallery * Update gallery/plot_optical_flow.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Fix import * Revert hardcoded normalization * fix uncommitted changes * Fix bug * Fix more bugs * Making resize optional for segmentation * Fixing preset * Fix mypy * Fixing documentation strings * Fix flake8 * minor refactoring Co-authored-by:
Anton Thomma <anton@pri.co.nz> Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Anton Thomma <11010310+thommaa@users.noreply.github.com> Co-authored-by:
-
- 24 Jan, 2022 1 commit
-
-
Yoshitomo Matsubara authored
* renamed ImageNet weights * renamed COCO weights * renamed COCO with VOC labels weights * renamed Kinetics 400 weights * rename default with DEFAULT * update test * fix typos * update test * update test * update test * indent as w was weight_enum * revert * Adding back the capitalization test Co-authored-by:
-
- 20 Jan, 2022 1 commit
-
-
Fedor authored
* Adding min_size as a required field. * Adding min_size to classification models (quantized and not) * Adding min_size to video models meta. * Moving min_size to _COMMON_META * Fixing extra line Co-authored-by:
-
- 19 Jan, 2022 1 commit
-
-
Philip Meier authored
-
- 06 Jan, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Improved meta-data for models. * Addressing comments from code-review. * Add parameter count. * Fix linter.
-
- 08 Dec, 2021 4 commits
-
-
Nicolas Hug authored
-
Nicolas Hug authored
Co-authored-by: -
Philip Meier authored
-
Vasilis Vryniotis authored
* Adding test that verifies meta schema. * Adding missing interpolation to keypoint model. * Change assertion. * Adding map metric in meta schema for detection and renaming metrics for kp and mask rcnn.
-
- 07 Dec, 2021 1 commit
-
-
Philip Meier authored
* simplify model builders * cleanup * refactor kwonly to pos or kw handling * put weight verification back * revert num categories checks * fix default weights * cleanup * remove manual parameter map * refactor decorator interface * address review comments * cleanup * refactor callable default * fix type annotation * process ungrouped models * cleanup * mroe cleanup * use decorator for detection models * add decorator for quantization models * add decorator for segmentation models * add decorator for video models * remove old helpers * fix resnet50 * Adding verification back on InceptionV3 * Add kwargs in DeeplabeV3 * Add kwargs on FCN * Fix typing on Deeplab * Fix typing on FCN Co-authored-by:
-
- 30 Nov, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Change the `default` weights mechanism to sue Enum aliases. * Change `get_weights` to work with full Enum names and make it public. * Applying improvements from code review.
-
- 29 Nov, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Rename classes Weights => WeightsEnum and WeightEntry => Weights. * Make enum values follow the naming convention `_V1`, `_V2` etc * Cleanup the Enum class naming conventions. * Add a test to check naming conventions.
-
- 27 Nov, 2021 1 commit
-
-
Yiwen Song authored
* [vit] Adding ViT to torchvision/models * adding pre-logits layer + resolving comments * Fix the model attribute bug * Change version to arch * fix failing unittests * remove useless prints * reduce input size to fix unittests * Increase windows-cpu executor to 2xlarge * Use `batch_first=True` and remove classifier * Change resource_class back to xlarge * Remove vit_h_14 * Remove vit_h_14 from __all__ * Move vision_transformer.py into prototype * Fix formatting issue * remove arch in builder * Fix type err in model builder * address comments and trigger unittests * remove the prototype import in torchvision.models * Adding vit back to models to trigger CircleCI test * fix test_jit_forward_backward * Move all to prototype. * Adopt new helper methods and fix prototype tests. * Remove unused import. Co-authored-by:
Vasilis Vryniotis <vvryniotis@fb.com>
-
- 26 Nov, 2021 1 commit
-
-
Philip Meier authored
* enable caching of model weights for prototype CI * syntax * syntax * make cache dir dynamic * increase verbosity * fix * use larget CI machine * revert debug output * [DEBUG] test env var usage in save_cache * retry * use checksum for caching * remove env vars because expansion is not working * syntax * cleanup * base caching on model-urls * relax regex * cleanup skips * cleanup * fix skipping logic * improve step name * benchmark without caching * benchmark with external download * debug * fix manual download location * debug again * download weights in the background * try parallel download * add missing import * use correct decoractor * up resource_class * fix wording * enable stdout passthrough to see download during test * remove linebreak * move checkout up * cleanup * debug failing test * temp fix * fix * cleanup * fix regex * remove explicit install of numpy
-
- 25 Nov, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Updated the link for densenet recipe. * Set default value of `num_classes` and `num_keypoints` to `None` * Provide helper methods for parameter checks to reduce duplicate code. * Throw errors on silent config overwrites from weight meta-data and legacy builders. * Changing order of arguments + fixing mypy. * Make the builders fully BC. * Add "default" weights support that returns always the best weights.
-
- 03 Nov, 2021 2 commits
-
-
Vasilis Vryniotis authored
* 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.
-
Vasilis Vryniotis authored
* Aligning exception with all other models. * Adding prototype preprocessing on video references. * Adding the rest of model builders on faster_rcnn.
-
- 02 Nov, 2021 1 commit
-
-
Vasilis Vryniotis authored
* 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.
-
- 28 Oct, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Add mutli-weight support for VideoResNet. * Fix linter. * Minor refactoring. * Update comments.
-
- 26 Oct, 2021 2 commits
-
-
Vasilis Vryniotis authored
* Adding multi-weight support to LRASPP * Adding tests for segmentation models. * Skip segmentation test by default.
-
Vasilis Vryniotis authored
* Refactoring tests. * Fixing lint. * Skip tests for models that don't have weights.
-
- 25 Oct, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Update model checkpoint for resnet50. * Add get_weight method to retrieve weights from name. * Update the references to support prototype weights. * Fixing mypy typing. * Switching to a python 3.6 supported equivalent. * Add unit-test. * Add optional num_classes.
-
- 20 Oct, 2021 2 commits
-
-
Joao Gomes authored
* adding Weights classes for Resnet classification models * Replacing BasicBlock by Bottleneck in all but 3 model contructors * adding tests for prototype models * fixing typo in environment variable * Update test/test_prototype_models.py Co-authored-by:
-
Philip Meier authored
* add CI job for prototype tests * register prototype job * fix torchvision installation * fix test invocation * ignore prototype tests in normal unittests Co-authored-by:
-
