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

tf2onnx模型目前还不支持指定输入的 shape 信息 #28

Closed
neiltian-tencent opened this issue Jun 9, 2020 · 2 comments
Closed
Assignees
Labels
enhancement New feature or request

Comments

@neiltian-tencent
Copy link
Collaborator

tf2onnx模型目前还不支持指定输入的 shape 信息,这对于大部分的 TF 模型没有问题,但是默写 tf 模型,在模型文件中并没有输入的 shape 信息,这就造成后面的转换失败。

@gttiankai gttiankai self-assigned this Jun 12, 2020
@gttiankai gttiankai added the enhancement New feature or request label Jun 12, 2020
@gttiankai
Copy link
Collaborator

目前在feature_align_model已经添加了相应的功能,如果 tf 转换TNN 模型失败,可以试试这个分支的代码,convert.md 文档中有如何使用。现在还有 docker image 没有更新,可以先使用 docker
build的方式来构建 docker image。

@gttiankai
Copy link
Collaborator

目前该功能已经merge到master分支。Docker image 也会自动化构建并上传docker hub.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants