torchvision/models/detection/rpn.py · 7d1cd1de34298ccb4a993d7c6af59b08cf8ac094 · OpenDAS / vision · GitLab

"vscode:/vscode.git/clone" did not exist on "8bbdc2db7ab4a1f08c2df81b517871d98368504c"

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