diff --git "a/Ascend-PyTorch\347\246\273\347\272\277\346\216\250\347\220\206\346\214\207\345\257\274/\344\270\223\351\242\230\346\241\210\344\276\213/\345\212\237\350\203\275\346\211\223\351\200\232/grid_sample\350\207\252\345\256\232\344\271\211onnx\347\256\227\345\255\220.md" "b/Ascend-PyTorch\347\246\273\347\272\277\346\216\250\347\220\206\346\214\207\345\257\274/\344\270\223\351\242\230\346\241\210\344\276\213/\345\212\237\350\203\275\346\211\223\351\200\232/grid_sample\350\207\252\345\256\232\344\271\211onnx\347\256\227\345\255\220.md" index 10b653903b8910320f0a0a2a2fc41658769030e5..965d0b1f81f08e2331d2f3c8349488bdc71c4a85 100644 --- "a/Ascend-PyTorch\347\246\273\347\272\277\346\216\250\347\220\206\346\214\207\345\257\274/\344\270\223\351\242\230\346\241\210\344\276\213/\345\212\237\350\203\275\346\211\223\351\200\232/grid_sample\350\207\252\345\256\232\344\271\211onnx\347\256\227\345\255\220.md" +++ "b/Ascend-PyTorch\347\246\273\347\272\277\346\216\250\347\220\206\346\214\207\345\257\274/\344\270\223\351\242\230\346\241\210\344\276\213/\345\212\237\350\203\275\346\211\223\351\200\232/grid_sample\350\207\252\345\256\232\344\271\211onnx\347\256\227\345\255\220.md" @@ -1,7 +1,8 @@ [PointRend](https://gitee.com/ascend/modelzoo/pulls/4611) +补丁说明: 1.onnx不支持grid_sample算子,参考mmcv的自定义算子grid_sample的测试等价代码bilinear_grid_sample进行替换 -2.去除部分动态shape的where算子 -3.expand输入类型转换 -4.div输入类型转换 -5.padv3算子问题影响网络精度,通过修改代码进行了规避 +2.去除部分om暂不支持的动态shape的where算子 +3.规避om expand算子暂不支持int64进行的输入类型转换 +4.规避om div算子暂不支持int64进行的输入类型转换 +5.om padv3算子问题影响网络精度,通过修改代码进行了规避 6.本模型是基于detectron2框架的模型 \ No newline at end of file