You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thanks for reporting the problem, the community uses https://discuss.tvm.ai/ as a way to collective resolve these questions. Please open a new troubleshooting thread there
When I convert a keras model to tvm, an error occur:
Traceback (most recent call last):
File "traffic_cnn_keras_tvm_dbg.py", line 225, in
mod, params = relay.frontend.from_keras(keras_model, shape_dict)
File "/home/yyh/3rdparty/tvm/python/tvm/relay/frontend/keras.py", line 821, in from_keras
zip_node = zip(node.node_indices, node.tensor_indices, node.inbound_layers)
TypeError: zip argument #1 must support iteration
My code:
`x = np.ones([1, 1, IMAGE_SIZE, IMAGE_SIZE])
shape_dict = {'conv2d_input': x.shape}
mod, params = relay.frontend.from_keras(keras_model, shape_dict)
compile the model
target = 'cuda'
ctx = tvm.gpu(0)
with relay.build_config(opt_level=3):
executor = relay.build_module.create_executor('graph', mod, ctx, target)
graph, lib, params = relay.build_module.build(mod, target, params=params)
lib.export_library("./deploy_lib.so")
print('lib export succeefully')
with open("./deploy_graph.json", "w") as fo:
fo.write(graph.json())
with open("./deploy_param.params", "wb") as fo:
fo.write(nnvm.compiler.save_param_dict(params))
`
The text was updated successfully, but these errors were encountered: