tests/test_nms.py · e1f6ef2c4cb8097e54ef1ae5d047800886bf519b · ModelZoo / SOLOv2-pytorch · GitLab

Find file Blame History Permalink

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

test_nms.py 2.24 KB