Hello, my goal is to simply run a CNN image recognition model built using TensorFlow on PYNQ FPGA using the tvm/vta stack.
More specifically, I’m working with the Inception V1 model found in this tutorial.
So far I have this:
with tf.gfile.FastGFile(model_path, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) graph = tf.import_graph_def(graph_def, name='') # call the utility to import the graph definition into default graph. graph_def = tf_testing.ProcessGraphDefParam(graph_def) # add shapes to the graph with tf.Session() as sess: graph_def = tf_testing.AddShapesToGraphDef(sess, 'softmax') sym, params = nnvm.frontend.from_tensorflow(graph_def) graph = nnvm.graph.create(sym) graph = vta.graph.clean_cast(graph) graph = vta.graph.clean_conv_fuse(graph)
which produces this error:
Traceback (most recent call last): File "tvm_classifier.py", line 134, in <module> generate_graph(graph_def) File "tvm_classifier.py", line 63, in generate_graph graph = vta.graph.clean_cast(graph) File "/home/youn/tvm/vta/python/vta/graph.py", line 208, in clean_cast elif op_name == "conv2d" and attrs["out_dtype"] == "int32": KeyError: 'out_dtype'
I’m a beginner so any help would be appreciated. I’m using TensorFlow version 1.6.0 If this is relevant (newest version also produces same error).