- 28 Aug, 2020 1 commit
-
-
vfdev authored
* [WIP] Unify ops Grayscale and RandomGrayscale * Unified inputs for grayscale op and transforms - deprecated F.to_grayscale in favor of F.rgb_to_grayscale * Fixes bug with fp input * [WIP] Updated code according to review * Removed unused import
-
- 26 Aug, 2020 1 commit
-
-
vfdev authored
* Improve transforms test codebase - refactored compareTensorToPIL, _create_data, approxEqualTensorToPIL methods * Fixed flake8
-
- 09 Jul, 2020 1 commit
-
-
mcarilli authored
* Fixes Xiao's repro * Ports nms to use full dispatcher * Move HIPGuard to nms_cuda * clang-format * run models in test_models.py on GPU if available * Francisco's comment, also disable cuda model tests to see if CPU alone still passes * cuda tests now pass locally, although still not comparing to saved numerics * add note for thing to ask francisco * Allow cuda and cpu tests to share a data file * ignore suffix if unneeded * Skip autocast numerics checks for a few models * Add roi_align test Co-authored-by:Michael Carilli <mcarilli@nvidia.com>
-
- 31 Mar, 2020 1 commit
-
-
Philip Meier authored
* remove six from python code * remove six from setup.py * remove six from tests * remove six from references * remove six from packaging * revert str to torch._six._string_classes * revert str to torch._six._string_classes
-
- 30 Nov, 2019 1 commit
-
-
driazati authored
* Add tests for results in script vs eager mode This copies some logic from `test_jit.py` to check that a TorchScript'ed model's outputs are the same as outputs from the model in eager mode. To support differences in TorchScript / eager mode outputs, an `unwrapper` function can be provided per-model. * Fix inception, use PYTORCH_TEST_WITH_SLOW * Update * Remove assertNestedTensorObjectsEqual * Add PYTORCH_TEST_WITH_SLOW to CircleCI config * Add MaskRCNN unwrapper * fix prec args * Remove CI changes * update * Update * remove expect changes * Fix tolerance bug * Fix breakages * Fix quantized resnet * Fix merge errors and simplify code * DeepLabV3 has been fixed * Temporarily disable jit compilation
-
- 01 Oct, 2019 1 commit
-
-
eellison authored
* add expected result tests * fix wrong assertion * start with only detection models * remove unneeded rng setting * fix test * add tuple support * update test * syntax error * treat .pkl files as binary data, see : https://git-scm.com/book/en/v2/Customizing-Git-Git-Attributes#_binary_files * fix test * fix elif * Map tensor results and enforce maximum pickle size * unrelated change * larger rtol * pass rtol atol around * last commit i swear... * respond to comments * fix flake * fix py2 flake
-
- 13 Jun, 2019 1 commit
-
-
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
-