- 11 May, 2020 1 commit
-
-
Philip Meier authored
* add mypy config * fix syntax error * fix annotations in torchvision/utils.py * add mypy type check to CircleCI * add mypy cache to ignore files * try fix CI * ignore flake8 F821 since it interferes with mypy * add mypy type check to config generator * explicitly set config files
-
- 07 Aug, 2019 1 commit
-
-
Edward Z. Yang authored
Following a similar line of inquiry to pytorch/audio#217
-
- 07 May, 2019 1 commit
-
-
Francisco Massa authored
* Initial layout for layers with cpp extensions * Move files around * Fix import after move * Add support for multiple types to ROIAlign * Different organization CUDA extensions work now * Cleanups * Reduce memory requirements for backwards * Replace runtime_error by AT_ERROR * Add nms test * Add support for compilation using CPP extensions * Change folder structure * Add ROIPool cuda * Cleanups * Add roi_pool.py * Fix lint * Add initial structures folder for bounding boxes * Assertion macros compatible with pytorch master (#540) * Support for ROI Pooling (#592) * ROI Pooling with tests. Fix for cuda context in ROI Align. * renamed bottom and top to follow torch conventions * remove .type().tensor() calls in favor of the new approach to tensor initialization (#626) * Consistent naming for rois variable (#627) * remove .type().tensor() calls in favor of the new approach to tensor initialization * Consistent naming for rois variable in ROIPool * ROIPool: Support for all datatypes (#632) * Use of torch7 naming scheme for ROIAlign forward and backward * use common cuda helpers in ROIAlign * use .options() in favor of .type() where applicable * Added tests for forward pass of ROIAlign, as well as more consistent naming scheme for CPU vs CUDA * working ROIAlign cuda backwards pass * working ROIAlign backwards pass for CPU * added relevant headers for ROIAlign backwards * tests for ROIAlign layer * replace .type() with .options() for tensor initialization in ROIAlign layers * support for Half types in ROIAlign * gradcheck tests for ROIAlign * updated ROIPool on CPU to work with all datatypes * updated and cleaned tests for ROI Pooling * Fix rebase problem * Remove structures folder * Improve cleanup and bugfix in test_layers * Update C++ headers * Add CUDAGuard to cu files * Add more checks to layers * Add CUDA NMS and tests * Add multi-type support for NMS CUDA * Avoid using THCudaMalloc * Add clang-format and reformat c++ code * Remove THC includes * Rename layers to ops * Add documentation and rename functions * Improve the documentation a bit * Fix some lint errors * Fix remaining lint inssues * Area computation doesn't add +1 in NMS * Update CI to use PyTorch nightly * Make NMS return indices sorted according to the score * Address reviewer comments * Lint fixes * Improve doc for roi_align and roi_pool * move to xenial * Fix bug pointed by @lopuhin * Fix RoIPool reference implementation in Python 2 Also fixes a bug in the clip_boxes_to_image -- this function needs a test! * Remove change in .travis
-
- 09 Apr, 2019 1 commit
-
-
Soumith Chintala authored
-
- 02 Apr, 2019 1 commit
-
-
Francisco Massa authored
* Add groups support to ResNet * Kill BaseResNet * Make it support multi-machine training
-
- 08 Jan, 2019 1 commit
-
-
Will Price authored
Test installed version of package To test against the installed version of the package (which is preferably since this can catch installation file inclusion bugs) we have to deal with quirks of coverage.py. By default the ecosystem is set up to do development coverage tests (e.g. in src, or by using `pip install -e .` or `python setup.py develop`). We want to test against the version installed, to do this we have to find the install path which we'll then pass to the `--cov` arg added by `pytest-cov`. To do this, we have to cd out of the current folder, and import the installed version and get it's install path (if we don't cd out, then we end up getting the existing directory in the cwd since by default cwd is on `sys.path`) Once we have the install path, we pass that to the `--cov` of pytest, however we also want to rewrite the paths for codecov to pick them up on the website, if they don't have to code coverage with local paths, they don't register is properly. In order to that, we have a .coveragerc file that has the contents: ``` [paths] source = torchvision /**/site-packages/torchvision ``` This tells codecov to treat all paths with those prefixes as the same, so anything like `/home/travis/miniconda/envs/test-environment/lib/python2.7/site-packages/torchvision/models/__init__.py` would be treated the same as `torchvision/models/__init__.py` after running the coverage combination command. The first path seems to be special, in that all subsequent paths are rewritten to that one. Once we collect coverage, we then run `coverage run` to rewrite the installation paths in the coverage file, `.coverage`, to those expected by codecov. Phew. And that's it. N.B: Whilst adding test/__init__.py does solve the issue, it also results in PWD being added to the path, and then we'll test the development version of the package, and not the installed version (see https://docs.pytest.org/en/latest/goodpractices.html#tests-as-part-of-application-code)
-
- 28 Jan, 2018 1 commit
-
-
Sanyam Kapoor authored
* Add basic Omniglot dataset loader * Remove unused import * Add Omniglot random pair to sample pair of characters * Precompute random set of pairs, deterministic after object instantiation * Export OmniglotRandomPair via the datasets module interfact * Fix naming convention, use sum instead of reduce * Fix downloading to not download everything, fix Python2 syntax * Fix end line lint * Add random_seed, syntax fixes * Remove randomized pair, take up as a separate generic wrapper * Fix master conflict
-
- 07 Oct, 2017 1 commit
-
-
Sasank Chilamkurthy authored
-
- 10 Nov, 2016 1 commit
-
-
Soumith Chintala authored
-