1. 12 Mar, 2021 1 commit
  2. 11 Mar, 2021 5 commits
  3. 10 Mar, 2021 7 commits
  4. 09 Mar, 2021 3 commits
  5. 08 Mar, 2021 2 commits
    • Nicolas Hug's avatar
      Speed up equalize transform: use bincount instead of histc (#3493) · 77e41870
      Nicolas Hug authored
      * use bincount instead of hist
      
      * only use bincount when on CPU
      
      * Added equality test for CPU vs cuda
      
      * Fix flake8 and tests
      
      * tuple instead of int for size
      77e41870
    • Nicolas Hug's avatar
      [OPS, IMP] New batched_nms implementation (#3426) · 414427dd
      Nicolas Hug authored
      
      
      * new batched_nms implem
      
      * flake8
      
      * hopefully fix torchscipt tests
      
      * Use where instead of nonzero
      
      * Use same threshold (4k) for CPU and GPU
      
      * Remove use of argsort
      
      * use views again
      
      * remove print
      
      * trying stuff, I don't know what's going on
      
      * previous passed onnx checks so the error isn't in _vanilla func. Trying to return vanilla now
      
      * add tracing decorators
      
      * cleanup
      
      * wip
      
      * ignore new path with ONNX
      
      * use vanilla if tracing...????
      
      * Remove script_if_tracing decorator as it was conflicting with _is_tracing
      
      * flake8
      
      * Improve coverage
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      414427dd
  6. 04 Mar, 2021 3 commits
  7. 03 Mar, 2021 3 commits
  8. 02 Mar, 2021 3 commits
  9. 01 Mar, 2021 13 commits