torchvision/models/detection/rpn.py · 66f2922028754c6a7af648e8d39c62a7d7a49659 · OpenDAS / vision · GitLab

Find file Blame History Permalink

Fix for AnchorGenerator when device switch happen (#1745) · 66f29220
Francisco Massa authored Jan 13, 2020
```
* Fix AnchorGenerator if moving from one device to another

* Fixes for the test
```
66f29220

rpn.py 19.5 KB