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

Issue with .pb to uff conversion #28

Open
sildipankar opened this issue Dec 28, 2020 · 1 comment
Open

Issue with .pb to uff conversion #28

sildipankar opened this issue Dec 28, 2020 · 1 comment

Comments

@sildipankar
Copy link

Following issue is observed:-
arning: No conversion function registered for layer: L2Norm_Helper_TRT yet.
Converting embeddings/Rsqrt as custom op: L2Norm_Helper_TRT
Traceback (most recent call last):
File "step01_pb_to_uff.py", line 33, in
output_filename=uff_file, text=False)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 225, in from_tensorflow
debug_mode=debug_mode)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 141, in convert_tf2uff_graph
uff_graph, input_replacements, debug_mode=debug_mode)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 126, in convert_tf2uff_node
op, name, tf_node, inputs, uff_graph, tf_nodes=tf_nodes, debug_mode=debug_mode)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 88, in convert_layer
fields = cls.parse_tf_attrs(tf_node.attr)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 269, in parse_tf_attrs
return {key: cls.parse_tf_attr_value(val) for key, val in attrs.items() if val is not None and val.WhichOneof('value') is not None}
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 269, in
return {key: cls.parse_tf_attr_value(val) for key, val in attrs.items() if val is not None and val.WhichOneof('value') is not None}
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 265, in parse_tf_attr_value
return cls.convert_tf2uff_field(code, val)
File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/converter.py", line 226, in convert_tf2uff_field
if isinstance(val, tf.AttrValue):
This was mentioned in #9 but I'm also using Jetpack 4.4 and it seem sthe project is updated for that.
Tensorflow Version 2.321 is used.

@jishang1602
Copy link

It works with tensorflow 1.15.0 or 1.15.2 version. please try with it

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

No branches or pull requests

2 participants