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.

Sign up for GitHub

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

Jump to bottom

先把checkpoint模型转为saved model（这个过程中没报错），再把saved model转onnx报错了 #151

Open
HUI11126 opened this issue Jan 13, 2021 · 3 comments
Open

先把checkpoint模型转为saved model（这个过程中没报错），再把saved model转onnx报错了 #151

HUI11126 opened this issue Jan 13, 2021 · 3 comments

Comments

@HUI11126
Copy link

HUI11126 commented Jan 13, 2021 •
edited
Loading

saved model转onnx时，报错：
ValueError: Node 'gradients/resnet_v1_101_4/block4/unit_3/bottleneck_v1/conv3/BatchNorm/FusedBatchNorm_grad/FusedBatchNormGrad' has an _output_shapes attribute inconsistent with the GraphDef for output #3: Dimension 0 in both shapes must be equal, but are 0 and 2048. Shapes are [0] and [2048].

请问，怎么解决？
@HUI11126 HUI11126 changed the title 转onnx报错了 先把checkpoint模型转为saved model（这个过程中没报错），再把saved model转onnx报错了 Jan 13, 2021
@HUI11126
Copy link
Author

checkpoint模型转为saved model是在tensorflow1.14版本转的，saved model转onnx是在tensorflow 2.2版本转的

@HUI11126
Copy link
Author

若是用tensorflow 1.14转saved model为onnx，则内存快速被占满，之后被杀死

@exmee
Copy link

exmee commented Nov 30, 2021

您好，请问怎么把checkpoint模型转为saved model呢，网络的输出节点output_node_names是什么呢，期待您的回复！

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@HUI11126 @exmee