1. 07 Aug, 2020 2 commits
    • vfdev's avatar
      [BC-breaking] Unified input for F.perspective (#2558) · 8c7e7bb0
      vfdev authored
      * [WIP] Added unified input perspective transformation code
      
      * Unified input for F.perspective
      - added tests
      - updated docs
      
      * Added more random test configs
      
      * Fixed the code according to PR's review
      8c7e7bb0
    • vfdev's avatar
      Unified inputs for `T.RandomRotation` (#2496) · 08af5cb5
      vfdev authored
      * Added code for F_t.rotate with test
      - updated F.affine tests
      
      * Rotate test tolerance to 2%
      
      * Fixes failing test
      
      * [WIP] RandomRotation
      
      * Unified RandomRotation with tests
      08af5cb5
  2. 06 Aug, 2020 1 commit
    • vfdev's avatar
      Fixes F.affine and F.rotate to support rectangular tensor images (#2553) · 025b71d8
      vfdev authored
      * Added code for F_t.rotate with test
      - updated F.affine tests
      
      * Rotate test tolerance to 2%
      
      * Fixes failing test
      
      * Optimized _expanded_affine_grid with a single matmul op
      
      * Recoded _compute_output_size
      
      * [WIP] recoded F_t.rotate internal methods
      
      * [WIP] Fixed F.affine to support rectangular images
      
      * Recoded _gen_affine_grid to optimized version ~ affine_grid
      - Fixes flake8
      
      * [WIP] Use _gen_affine_grid for affine and rotate
      
      * Fixed tests on square / rectangular images for affine and rotate ops
      
      * Removed redefinition of F.rotate
      - due to bad merge
      025b71d8
  3. 05 Aug, 2020 1 commit
    • vfdev's avatar
      Unified inputs for `F.rotate` (#2495) · 76662528
      vfdev authored
      * Added code for F_t.rotate with test
      - updated F.affine tests
      
      * Rotate test tolerance to 2%
      
      * Fixes failing test
      
      * Optimized _expanded_affine_grid with a single matmul op
      
      * Recoded _compute_output_size
      76662528
  4. 04 Aug, 2020 1 commit
  5. 03 Aug, 2020 13 commits
  6. 31 Jul, 2020 9 commits
  7. 30 Jul, 2020 2 commits
  8. 29 Jul, 2020 2 commits
  9. 20 Jul, 2020 1 commit
  10. 17 Jul, 2020 1 commit
  11. 16 Jul, 2020 1 commit
    • vfdev's avatar
      Unified input for F.affine (#2444) · 5f4b5794
      vfdev authored
      * [WIP] F.affine
      
      * [WIP] F.affine + tests
      
      * Unified input for F.affine
      
      * Removed commented code
      
      * Removed unused imports
      5f4b5794
  12. 15 Jul, 2020 2 commits
  13. 12 Jul, 2020 2 commits
  14. 09 Jul, 2020 2 commits
    • Francisco Massa's avatar
      Fix wrong clamping in RoIAlign with aligned=True (#2438) · 131ba132
      Francisco Massa authored
      * Fix wrong clamping in RoIAlign with aligned=True
      
      * Fix silly mistake
      
      * Bugfix pointed out during code-review
      131ba132
    • 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