Got an error with from_tensorflow.py in the tutorials #1971
Description
Hi all,
I have just started using TVM. However, after installing TVM, I got an error in running the script from_tensorflow.py in the tutorials :
Traceback (most recent call last):
File "from_tensorflow.py", line 112, in
sym, params = nnvm.frontend.from_tensorflow(graph_def, layout=layout)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1371, in from_tensorflow
sym, params = g.from_tensorflow(graph, layout)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1145, in from_tensorflow
op = self._convert_operator(node.op, inputs, attr, graph)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 1332, in _convert_operator
sym = convert_map[op_name](inputs, attrs, self._params)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 309, in _impl
return AttrCvt(op_name='cast', transforms={'DstT': 'dtype'}, ignores=['SrcT'])(inputs, attr)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/tensorflow.py", line 46, in call
self._custom_check)(inputs, attrs, *args)
File "/home/hung/projects/tvm/nnvm/python/nnvm/frontend/common.py", line 107, in call
return get_nnvm_op(op_name)(*inputs, **new_attrs)
File "/home/hung/projects/tvm/nnvm/python/nnvm/_ctypes/symbol.py", line 181, in creator
ctypes.byref(sym_handle)))
File "/home/hung/projects/tvm/nnvm/python/nnvm/_base.py", line 75, in check_call
raise NNVMError(py_str(_LIB.NNGetLastError()))
nnvm._base.NNVMError: Cannot find argument 'Truncate', Possible Arguments:
dtype : {'float16', 'float32', 'float64', 'int16', 'int32', 'int64', 'int8', 'uint16', 'uint32', 'uint64', 'uint8'}, required
Output data type.
, in operator cast(name="", dtype="float32", Truncate="False")
I am using Ubuntu 16.04, Tensorflow 1.11.0 with cpu. Any idea to fix this bug would be greatly appreciated!
Best,
Hung