Change input to data type

apache · Feb 18, 2020 · 57870ee · 57870ee
1 parent a22c040
commit 57870ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
@@ -614,7 +614,7 @@ def _convert_data_type(input_type):
     elif input_type in ["byte", "torch.uint8"]:
         return "uint8"
     else:
-        raise NotImplementedError("input_type {} is not handled yet" % (data_type))
+        raise NotImplementedError("input_type {} is not handled yet" % (input_type))
     return "float32"
 
 def _create_typed_const(data, data_type):