tests/test_ops/test_onnx.py · 72e4cc12bd3f6aa882183ad78c613d12af929755 · OpenDAS / MMCV · GitLab

Find file Blame History Permalink

Fix pytorch2onnx failed for interpolate op with PyTorch==1.6.0(mmdet#4646) (#848) · 72e4cc12
RunningLeon authored Feb 26, 2021
```
* Fix pytorch2onnx for yolov3 with torch==1.6.0

* update and add test for F.interpolate

* add test_onnx.py with build_cuda
```
72e4cc12

test_onnx.py 11.5 KB