Hi, everyone, I want to teach NNVM recognize tensorflow model. I want to know anyone knows the tensorflow model specification like ONNX (https://github.com/onnx/onnx/blob/master/docs/IR.md) and NVVM’s model specification? Morever, anyone knows when to convert tensorflow model to NVVM, anythings I need to take care? Are there any existing tests I could leverage to make sure my conversion is right?
Thanks for replying. I have tested ONNX, but it seems that it is not mature, I convert mobilenet model occurred error(https://github.com/onnx/tensorflow-onnx/issues/7). So I think support tf to nnvm directly is good option.
For testing, thanks for the advice, I will leverage it in my implementation.
Thanks for the information. I noticed that but it just list overview of tf model, doesn’t list more detail information such as op like ONNX / NNVM does.
Did you encounter problems in converting resnet50 tf model to coreml? I had error messages as https://github.com/tf-coreml/tf-coreml/issues/210, but I saw you reported the error message on a different model. Any advice on how to solve the resnet50 conversion issue? Thanks a lot.
Did you use v1 or v2 on that page? How did you freeze it?
I downloaded v1, and used python freeze_graph.py --input_graph=resnet_v1_50_inf_graph.pb --input_checkpoint=resnet_v1_50.ckpt --input_binary=true --output_graph=frozen_resnet_v1_50_slim.pb --output_node_names=resnet_v1_50/predictions/Reshape_1
to freeze it. But I got an error:
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py", line 1759, in restore
err, "a mismatch between the current graph and the graph")
tensorflow.python.framework.errors_impl.InvalidArgumentError: Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:
Assign requires shapes of both tensors to match. lhs shape= [1,1,2048,1001] rhs shape= [1,1,2048,1000]