- 11 Jun, 2019 1 commit
-
-
Tomas Alori authored
Changes made in 7716aba5 broke calls to this method.
-
- 07 Jun, 2019 4 commits
-
-
Francisco Massa authored
-
Francisco Massa authored
* GPU efficient Densenets * removed `import math` * Changed 'efficient' to 'memory_efficient' * Add tests * Bugfix in test * Fix lint * Remove unecessary formatting
-
Matthew Yeung authored
* allow user to define residual settings * 4spaces * linting errors * backward compatible, and added test
-
peterjc123 authored
-
- 06 Jun, 2019 1 commit
-
-
Vinh Nguyen authored
* adding mixed precision training with Apex * fix APEX default optimization level * adding python version check for apex * fix LINT errors and raise exceptions if apex not available
-
- 05 Jun, 2019 3 commits
-
-
Soumith Chintala authored
-
Surgan Jandial authored
* updating docs for randomperspective * my * ci
-
hirune924 authored
-
- 03 Jun, 2019 3 commits
-
-
Brian Johnson authored
-
Philip Meier authored
* added fake data * fixed fake data * renamed extract and download methods and added functionality * added raw fake data * refactored imagenet and added test * flake8 * added fake devkit and mocked download_url * reversed uncommenting * added mock to CI * fixed tests for imagefolder * flake8
-
Gu-ni-kim authored
Add 'import torch' in example
-
- 31 May, 2019 1 commit
-
-
Ailing authored
* add more hub models * fix lint
-
- 30 May, 2019 1 commit
-
-
Soumith Chintala authored
* make C extension lazy-import * add lazy loading to roi_pool
-
- 29 May, 2019 3 commits
-
-
Edouard Oyallon authored
* modif of the STL10 loader * missing space
-
Francisco Massa authored
* Fix STL10 repr * Do not inherit from Cifar10 * Make it safer to inherit from VisionDataset
-
Francisco Massa authored
* WIP * WIP: minor improvements * Add tests * Fix typo * Use download_and_extract on caltech, cifar and omniglot * Add a print message during extraction * Remove EMNIST from test
-
- 27 May, 2019 2 commits
-
-
Tongzhou Wang authored
-
Francisco Massa authored
* add USPS dataset * minor fixes * Improvements to the USPS dataset Add it to the documentation, expose it to torchvision.datasets and inherit from VisionDataset
-
- 25 May, 2019 1 commit
-
-
7d authored
Consider the difference of the division operator between Python 2.x and Python 3.x.
-
- 24 May, 2019 1 commit
-
-
Konstantin Lopuhin authored
* Allow forcing GPU build with FORCE_CUDA=1 This is convenient to e.g. build with GPU support inside a docker image * Document FORCE_CUDA
-
- 23 May, 2019 3 commits
-
-
Francisco Massa authored
* #944 MSBuild Compile time casting Error * #944 MSBuild Error static_cast<Long> to static_cast<int64_t> * Add eval.py Not Work find_contours * Remove unnecessary file * Lint
-
peterjc123 authored
-
Varun Agrawal authored
Updated nms_cuda signature to accept detections and scores as separate tensors. This also required updating the indexing in the NMS CUDA kernel. Also made the iou_threshold parameter name consistent across implementations.
-
- 22 May, 2019 8 commits
-
-
Soumith Chintala authored
-
Soumith Chintala authored
This reverts commit 06b9b0fc.
-
Francisco Massa authored
-
Sergey Zagoruyko authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
- 21 May, 2019 8 commits
-
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
Allows for easily evaluating the pre-trained models in the modelzoo
-
Francisco Massa authored
This makes it consistent with the other models, which returns nouns in plurial
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
Also adds documentation for the segmentation models
-