Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

custom_ops映射使用问题 #1245

Open
FrankyTang opened this issue May 7, 2024 · 8 comments
Open

custom_ops映射使用问题 #1245

FrankyTang opened this issue May 7, 2024 · 8 comments
Assignees
Labels
Bug Something isn't working stale

Comments

@FrankyTang
Copy link

我在使用命令行进行put_along_axis映射,使用了custom_ops,但是onnx的输入和输出发现对不上
这个是我使用的命令行:paddle2onnx --model_dir . --model_filename paddle_put_along_axis.pdmodel --opset_version 16 --save_file put_along_axis.onnx --custom_ops '{"put_along_axis":"ScatterElements"}'

下面的图片可以看到输入的名称和变量都对不上,可能是我使用的不对,能否提供一个正确的使用教程

image

@Zheng-Bicheng Zheng-Bicheng added the Bug Something isn't working label May 17, 2024
@Zheng-Bicheng
Copy link
Collaborator

@Jiang-Jia-Jun 这个参数还有在使用吗?

@Zheng-Bicheng
Copy link
Collaborator

这个好像没错呀,你这个输入变量名称原本应该是什么呢?

@Zheng-Bicheng
Copy link
Collaborator

custom_ops映射我们移除了,这个确实有问题,请问是在哪个地方需要使用到这个东西呢?

@Zheng-Bicheng
Copy link
Collaborator

现在的Runtime都是和ONNX的标准对齐的,理论上用不到这个功能了

@FrankyTang
Copy link
Author

我的场景是paddle->onnx->trt。然后模型涉及到自定义算子,我就需要使用custom_ops完成paddle->onnx的转换。

@Zheng-Bicheng
Copy link
Collaborator

@FrankyTang 您是自己使用trt实现了算子是吗?所以需要用CustomOps?

@FrankyTang
Copy link
Author

@Zheng-Bicheng 是的,我有自定义的trt算子。

@Zheng-Bicheng Zheng-Bicheng reopened this Nov 14, 2024
Copy link

This issue is stale because it has been open for 30 days with no activity.

@github-actions github-actions bot added the stale label Dec 15, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Bug Something isn't working stale
Projects
None yet
Development

No branches or pull requests

3 participants