- 14 Jun, 2019 1 commit
-
-
François Darmon authored
Change documentation of perspective(). The doc was about an old version that used directly transformation parameters
-
- 13 Jun, 2019 4 commits
-
-
Francisco Massa authored
* Raise error during downloading * Fix py2 error and lint
-
Philip Meier authored
-
Philip Meier authored
* create imagenet fakedata on-the-fly * flake8 * Minor test refactorings (#1011) * Make tests work on fbcode * Lint * Fix rebase error * Properly use get_file_path_2 * Fix wrong use of get_file_path_2 again * Missing import * create imagenet fakedata on-the-fly
-
Francisco Massa authored
* Make tests work on fbcode * Lint * Fix rebase error * Properly use get_file_path_2 * Fix wrong use of get_file_path_2 again * Missing import
-
- 12 Jun, 2019 2 commits
-
-
Philip Meier authored
* added test for CIFAR10(0) * create CIFAR data on-the-fly * flake8 * fixed typo * removed falsely added import
-
Leon Bottou authored
* Added general reader for sn3 tensors in "pascalvincent" format * Added class QMNIST into mnist.py * QMNIST dataset: make some pt files smaller * Change request from fmassa. * read_sn3_pascalvincent_tensor: cse * read_sn3_pascalvincent_tensor: check file size (when strict!=False) * Fix lint * More lint * Add documentation and expose QMNIST to dataset namespace
-
- 11 Jun, 2019 4 commits
-
-
ekka authored
This PR uses a protected method for loading and initializing the segmentation models. Relevant #875
-
Dougal J. Sutherland authored
* CelebA: track attr names, support split="all", code cleanup * fix typo
-
Shahriar authored
* Added the existing code * Added squeezenet and fixed some stuff in the other models * Wrote DenseNet and a part of InceptionV3 Going to clean and check all of the models and finish inception * Fixed some errors in the models Next step is writing inception and comparing with python code again. * Completed inception and changed models directory * Fixed and wrote some stuff * fixed maxpoool2d and avgpool2d and adaptiveavgpool2d * Fixed a few stuff Moved cmakelists to root and changed the namespace to vision and wrote weight initialization in inception * Added models namespace and changed cmakelists the project is now installable * Removed some comments * Changed style to pytorch style, added some comments and fixed some minor errors * Removed truncated normal init * Changed classes to structs and fixed a few errors * Replaced modelsimpl structs with functional wherever possible * Changed adaptive average pool from struct to function * Wrote a max_pool2d wrapper and added some comments * Replaced xavier init with kaiming init * Fixed an error in kaiming inits * Added model conversion and tests * Fixed a typo in alexnet and removed tests from cmake * Made an extension of tests and added module names to Densenet * Added python tests * Added MobileNet and GoogLeNet models * Added tests and conversions for new models and fixed a few errors * Updated Alexnet ad VGG * Updated Densenet, Squeezenet and Inception * Added ResNexts and their conversions * Added tests for ResNexts * Wrote tools nessesary to write ShuffleNet * Added ShuffleNetV2 * Fixed some errors in ShuffleNetV2 * Added conversions for shufflenetv2 * Fixed the errors in test_models.cpp * Updated setup.py * Fixed flake8 error on test_cpp_models.py * Changed view to reshape in forward of ResNet * Updated ShuffleNetV2 * Split extensions to tests and ops * Fixed test extension * Fixed image path in test_cpp_models.py * Fixed image path in test_cpp_models.py * Fixed a few things in test_cpp_models.py * Put the test models in evaluation mode * Fixed registering error in GoogLeNet * Updated setup.py * write test_cpp_models.py with unittest * Fixed a problem with pytest in test_cpp_models.py * Fixed a lint problem
-
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 6 commits
-
-
Soumith Chintala authored
-
Soumith Chintala authored
This reverts commit 06b9b0fc.
-
Francisco Massa authored
-
Sergey Zagoruyko authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-