1. 08 Mar, 2021 1 commit
    • 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
  2. 04 Mar, 2021 3 commits
  3. 03 Mar, 2021 3 commits
  4. 02 Mar, 2021 3 commits
  5. 01 Mar, 2021 17 commits
  6. 26 Feb, 2021 2 commits
  7. 25 Feb, 2021 4 commits
  8. 24 Feb, 2021 1 commit
  9. 23 Feb, 2021 5 commits
  10. 22 Feb, 2021 1 commit