Skip to content

T
tic
Commit a748f5f3 by Howave Committed by Tianqi Chen
Options

[nnvm] fix nnvm compiler build module error (#3378)

parent 04e81624
Showing with 1 additions and 1 deletions
nnvm/python/nnvm/compiler/build_module.py
...@@ -148,7 +148,7 @@ def _update_shape_dtype(shape, dtype, params): ...@@ -148,7 +148,7 @@ def _update_shape_dtype(shape, dtype, params):
shape.update({k : v.shape for k, v in params.items()}) shape.update({k : v.shape for k, v in params.items()})
if isinstance(dtype, str): if isinstance(dtype, str):
for k, v in params.items(): for k, v in params.items():
if v.dtype != dtype: if v.dtype != dtype and v.shape:
raise ValueError( raise ValueError(
"%s: dtype not expected %s vs %s" % (k, dtype, v.dtype)) "%s: dtype not expected %s vs %s" % (k, dtype, v.dtype))
else: else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment