1. 28 Aug, 2020 1 commit
    • vfdev's avatar
      Unified inputs for grayscale ops and transforms (#2586) · 2eba1f04
      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
      2eba1f04
  2. 26 Aug, 2020 1 commit
  3. 09 Jul, 2020 1 commit
    • mcarilli's avatar
      [WIP] Allow autocast for 1.6 (#2384) · 0a8586c9
      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: default avatarMichael Carilli <mcarilli@nvidia.com>
      0a8586c9
  4. 31 Mar, 2020 1 commit
    • Philip Meier's avatar
      Remove six dependency (#2017) · 42b8d462
      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
      42b8d462
  5. 30 Nov, 2019 1 commit
    • driazati's avatar
      Add tests for results in script vs eager mode (#1430) · 227027d5
      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
      227027d5
  6. 01 Oct, 2019 1 commit
    • eellison's avatar
      Add expected result tests (#1377) · 96ec0e1d
      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
      96ec0e1d
  7. 13 Jun, 2019 1 commit
    • Francisco Massa's avatar
      Minor test refactorings (#1011) · e402d43f
      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
      e402d43f