Commits · e1f6ef2c4cb8097e54ef1ae5d047800886bf519b · ModelZoo / SOLOv2-pytorch

07 Nov, 2019 1 commit
- Add tests for NMS / Issue with NMS on GPU 1. (#1603) · e1f6ef2c
  Jon Crall authored Nov 07, 2019
```
* Add tests for NMS

* Fix linting errors

* Add DeviceGaurd to nms_cuda
```
  e1f6ef2c
28 Jul, 2019 1 commit
- use .scalar_type() instead of .type() to suppress some warnings (#1070) · 8387aba8
  Kai Chen authored Jul 28, 2019
  
  8387aba8
07 Apr, 2019 1 commit
- port nms extension from maskrcnn-benchmark (#469) · e421e832
  Kai Chen authored Apr 06, 2019
```
* port nms extension from maskrcnn-benchmark

* fix linting error
```
  e421e832