Dmitry Sidnev
|
a88d1d28c1
|
[Feature] enable exporting to onnx for PointRend (#953)
* Fix export to onnx for PointRend
* Fix codestyle
* Fix codestyle
* Fix type in docstring
* Minor fix
* Fix export with custom ops
* Fix codestyle
* Add tests for bilinear_grid_sample function
* Remove redundant operation and rename variables
* Fix bug in bilinear_grid_sample and update test
* Fix getting batch size
* skip torch==1.3.1
* remove unused import
* fix lint
* support export with batch
* fix dynamic clip
* skip test for torch<1.5.0
* Add docstrings and comments
* Minor fix
* Recover clipping code
* Fix clamping in pytorch 1.7.0
* Fix bilinear_grid_sampler
* Minor fix
Co-authored-by: maningsheng <maningsheng@sensetime.com>
|
2021-06-11 13:49:19 +08:00 |