"src/vscode:/vscode.git/clone" did not exist on "2755ef62d97a97d76443d7d07125629a7d6c41bc"
- 28 Oct, 2021 2 commits
-
-
Vasilis Vryniotis authored
* Add mutli-weight support for VideoResNet. * Fix linter. * Minor refactoring. * Update comments.
-
Jirka Borovec authored
Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 27 Oct, 2021 4 commits
-
-
Joao Gomes authored
* adding multiweight support for deeplabv3 prototype models * adding default values for optional params * fixing bug * addressing PR comment
-
Nicolas Hug authored
* Change test_random_apply * Change test_random_choice * Change test_randomness * took care of RandomVert/HorizFlip * take care of RandomGrayScale * minor cleanup * avoid 0 degree rotation just in case
-
Vasilis Vryniotis authored
* Fix flakiness on the TestStochasticDepth test. * Fix minor bug when p=1.0 * Remove device and dtype setting.
-
Philip Meier authored
* cache dataset info * try loading categories file by default Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 26 Oct, 2021 6 commits
-
-
Vasilis Vryniotis authored
* Adding multi-weight support to LRASPP * Adding tests for segmentation models. * Skip segmentation test by default.
-
David Fan authored
Co-authored-by:Prabhat Roy <prabhatroy@fb.com>
-
Philip Meier authored
* replace np.frombuffer with torch.frombuffer in MNIST prototype * cleanup * appease mypy * more cleanup * clarify inplace offset * fix num bytes for floating point data
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/4620 Pull Request resolved: https://github.com/pytorch/nestedtensor/pull/455 Fixes missing kernel guards as identified by D30072495 Reviewed By: jingsh, xush6528 Differential Revision: D31553158 fbshipit-source-id: 80de017ba2ddc52e2a684df9b3eae5de84ed49f4 Co-authored-by:
Richard Barnes <rbarnes@fb.com>
-
Vasilis Vryniotis authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/4571 Fixing windows build for `torchvision` when `MOBILE` is defined. In `csrc/vision.cpp`, since `PyMODINIT_FUNC` depends on `Python.h` I added the same condition for `PyMODINIT_FUNC` as the one for `import <PyTorch.h>`. Reviewed By: malfet Differential Revision: D31488734 fbshipit-source-id: fd52a9ab161288d9c8bf3b9e0ab9da856a4ebadb Co-authored-by:
Mengwei Liu <larryliu@fb.com>
-
- 25 Oct, 2021 6 commits
-
-
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.
-
Francisco Massa authored
Summary: As titled. Reviewed By: sallysyw Differential Revision: D31813675 fbshipit-source-id: 8ff879be172530bcc41aae5d71c67e1e90f09cbf Co-authored-by:Kai Zhang <kaizh@fb.com>
-
Francisco Massa authored
* Expand usage logging to detection models Summary: Track all detection model instantiation. Reviewed By: sallysyw Differential Revision: D31741603 fbshipit-source-id: 89467db5d89ea41c5a9ab95d1846264d0a1f4199 * Fix lint Co-authored-by: -
Francisco Massa authored
Summary: We would like to track the internal usage of TorchVision components. We're tracking [datasets usage](https://fburl.com/daiquery/dqpmemn3 ) now. This diff expand the tracking to all models. Reviewed By: fmassa Differential Revision: D31441632 fbshipit-source-id: e26072e582ac9f832c2056307ebf0eccf2ed6c9c Co-authored-by:
-
Nicolas Hug authored
-
Vasilis Vryniotis authored
* Add mutli-weight support for FCN. * Rename base_size to resize_size.
-
- 22 Oct, 2021 5 commits
-
-
Joao Gomes authored
* Adding multiweight support for mnasnet prototype * Adressing PR comments * adding return type to _mnasnet Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
Joao Gomes authored
* Adding multiweight support for mobilenetv3 prototype
-
Vasilis Vryniotis authored
* Update EMA every X iters. * Adding AdamW optimizer. * Adjusting EMA decay scheme. * Support custom weight decay for Normalization layers. * Fix identation bug. * Change EMA adjustment. * Quality of life changes to faciliate testing * ufmt format * Fixing imports. * Adding FixRes improvement. * Support EMA in store_model_weights. * Adding interpolation values. * Change train_crop_size. * Add interpolation option. * Removing hardcoded interpolation and sizes from the scripts. * Fixing linter. * Incorporating feedback from code review.
-
Joao Gomes authored
* adding efficientnet to prototype * Removing _efficientnet_conf from prototype and main efficientnet classification models * fixing merge conflicts * fixing merge conflicts * fixing lint errors * fixing lint errors * fixing lint errors
-
Joao Gomes authored
* adding efficientnet to prototype * renaming _efficientnet_model in torchvision.model to be aligned with prototype implementation * adding specific cop sizes to ech model * adding specific cop and resize transforms * fixing lint errors * Update torchvision/prototype/models/efficientnet.py Co-authored-by:
-
- 21 Oct, 2021 7 commits
-
-
Joao Gomes authored
* adding interpolation model information to all classification prototype models * fixing lint errors Co-authored-by: -
Sergii Khomenko authored
* Migrate mnist dataset from np.frombuffer * Add a copy with bytearray for non-writable buffers * Add byte reversal for mnist Co-authored-by: -
Philip Meier authored
* make mypy more strict for prototype datasets * fix code format * apply strictness only to datasets * fix more mypy issues * cleanup * fix mnist annotations * refactor celeba * warn on redundant casts * remove redundant cast * simplify annotation * fix import
-
Nicolas Hug authored
* WIP * cleaner code * Add tests * Add docs * Assert dtype * put back check * Address comments Co-authored-by: -
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: -
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 2 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:
-
