"vscode:/vscode.git/clone" did not exist on "eebd9767155163289114e58bfd6454bd9d524cc4"
- 21 Oct, 2021 4 commits
-
-
Nicolas Hug authored
* WIP * cleaner code * Add tests * Add docs * Assert dtype * put back check * Address comments Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
* add prototype imagenet dataset * add missing checksums * fix mypy * add human readable categories * cleanup * sort categories ascending based on wnid * remove accidentally added file * cleanup category file generation * fix mypy
-
Nicolas Hug authored
-
Philip Meier authored
* fix category file inclusion * fix deprecated import
-
- 20 Oct, 2021 8 commits
-
-
Joao Gomes authored
* Densenet121 added * All densenet prototypes added * fixing flake8 errors * fixing argument type
-
Joao Gomes authored
* adding vgg 11 * adding remaining vgg prototype models * fixing path for vgg16_bn molde Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Joao Gomes authored
* adding alexnet prototype model * adding recipe reference * fixing lint Co-authored-by: -
Philip Meier authored
-
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:
-
Vasilis Vryniotis authored
* Refactoring resnet_fpn backbone building. * Passing the change to *_rcnn and retinanet. * Applying for faster_rcnn + mobilenetv3 * Applying for ssdlite + mobilenetv3 * Applying for ssd + vgg16 * Update the expected file of retinanet_resnet50_fpn to fix order of initialization. * Adding full model weights for the VGG16 features.
-
Prabhat Roy authored
* Fixed unused variables in ops * Fixed test failure
-
Philip Meier authored
-
- 19 Oct, 2021 5 commits
-
-
Vasilis Vryniotis authored
* Move FCN methods to itsown package. * Fix lint. * Move LRASPP methods to their own package. * Move DeepLabV3 methods to their own package. * Adding deprecation warning for torchvision.models.segmentation.segmentation. * Refactoring deeplab. * Setting aux default to false. * Fixing imports. * Passing backbones instead of backbone names to builders. * Fixing mypy * Addressing review comments. * Correcting typing. * Restoring special handling for references.
-
Philip Meier authored
* streamline category file generation for prototype datasets * cleanup Co-authored-by:Prabhat Roy <prabhatroy@fb.com>
-
Aditya Oke authored
* Start adding types * add typing * Type prototype models * fix optional type bug * transient import * Fix weights type * fix import * Apply suggestions from code review Address nits Co-authored-by:
Khushi Agrawal <khushiagrawal411@gmail.com> Co-authored-by:
-
Nicolas Hug authored
-
Philip Meier authored
-
- 18 Oct, 2021 6 commits
-
-
Philip Meier authored
-
Bruno Korbar authored
* initial commit * initial commit * flake8 * last nits Co-authored-by: -
Vasilis Vryniotis authored
* Fixing minor issue on typing. * Sample implementation for quantized resnet50.
-
Vasilis Vryniotis authored
-
Philip Meier authored
Co-authored-by: -
Guo authored
We no longer cache the data in a custom binary, but simply read from the raw data directly. Co-authored-by:Philip Meier <github.pmeier@posteo.de>
-
- 15 Oct, 2021 3 commits
-
-
Kai Zhang authored
-
Aditya Oke authored
* Annotate rpn * Small fix * Small fix and ignore Co-authored-by: -
Vasilis Vryniotis authored
* Adding FasterRCNN ResNet50. * Refactoring to remove duplicate code. * Adding typing info. * Setting weights_backbone=None as default value. * Overwrite eps only for specific weights.
-
- 14 Oct, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Adding lightweight API for models. * Adding resnet50. * Fix preset * Add fake categories. * Fixing mypy. * Add string=>weight conversion support on Enums. * Temporarily hardcoding imagenet categories. * Minor refactoring.
-
- 13 Oct, 2021 2 commits
-
-
Muhammed Abdullah authored
* Added Dropout parameter of Models * Added argument description for dropout in MobileNet v2 and v3 Updated quantization/googlenet.py as per the changes in constructor in googlenet * Moved new dropout parameter n the end * Updated googlenet.py Co-authored-by: -
Jirka Borovec authored
Co-authored-by:
deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com> Co-authored-by:
-
- 12 Oct, 2021 2 commits
-
-
Aditya Oke authored
* Add typing for imagelist, move params to docstring * Add docs
-
Aditya Oke authored
* Start annotating utils * checking * Add annotations at _utils.py * Remove unnecessary comments. * re-checked typings * Update typing * Ignore small error * Use optional tensor * Ignore for JIT Co-authored-by:
-
- 11 Oct, 2021 1 commit
-
-
Vasilis Vryniotis authored
-
- 08 Oct, 2021 3 commits
-
-
Sergii Khomenko authored
Co-authored-by: -
Philip Meier authored
Co-authored-by: -
Joao Gomes authored
* mExplicitly copying array in pil_to_tensor * Update torchvision/transforms/functional.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Adding comments regarding implicit array deep copy in PILToTensor transform * Update torchvision/transforms/transforms.py Co-authored-by:
-
- 07 Oct, 2021 5 commits
-
-
Philip Meier authored
* add benchmark utility for datasets * cleanup * cleanup * add benchmarks for MNIST and variants * simplifiy return if exeception * refactor benchmark instantiation logic Co-authored-by: -
vfdev authored
* Use FX feature extractor for segm model * Removed use_fe option
-
Philip Meier authored
Co-authored-by: -
Philip Meier authored
-
Philip Meier authored
Co-authored-by:
-
