1. 30 Mar, 2021 5 commits
  2. 25 Mar, 2021 1 commit
  3. 24 Mar, 2021 2 commits
    • Philip Meier's avatar
      add tests for MNIST and variants (#3423) · aa365993
      Philip Meier authored
      * add tests for MNIST and variants
      
      * remove old tests and fakedata generation
      
      * fix default config detection for if dataset has variable keywords
      
      * use split="mnist" as default for EMNIST
      
      * fix QMNIST tests
      
      * lint
      
      * fix special kwargs detection
      
      * Revert "use split="mnist" as default for EMNIST"
      
      This reverts commit 62c9b23597f4a391cad409cbd93edac1b3474acf.
      
      * fix tests
      
      * fix QMNIST test case name
      
      * remove dead code from test
      
      * Revert "remove old tests and fakedata generation"
      
      This reverts commit a285b97c4827566a5bc06c288f5bba8d614a4f7a.
      
      * remove old tests
      
      * readd removed import
      aa365993
    • Philip Meier's avatar
      Improve error handling in make_dataset (#3496) · 0818c682
      Philip Meier authored
      * factor out find_classes
      
      * use find_classes in video datasets
      
      * adapt old tests
      0818c682
  4. 22 Mar, 2021 2 commits
  5. 20 Mar, 2021 1 commit
  6. 19 Mar, 2021 1 commit
  7. 17 Mar, 2021 2 commits
  8. 16 Mar, 2021 2 commits
  9. 15 Mar, 2021 2 commits
  10. 12 Mar, 2021 3 commits
  11. 11 Mar, 2021 2 commits
  12. 10 Mar, 2021 4 commits
  13. 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
  14. 04 Mar, 2021 3 commits
  15. 03 Mar, 2021 3 commits
  16. 02 Mar, 2021 2 commits
  17. 01 Mar, 2021 3 commits