From 57870ee4347f9d9b08b2d06975742d9a4da1c11c Mon Sep 17 00:00:00 2001 From: alexwong <11878166+alexwong@users.noreply.github.com> Date: Tue, 18 Feb 2020 12:45:21 -0800 Subject: [PATCH] Change input to data type --- python/tvm/relay/frontend/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index 0954bdb726643..d2c680188f99c 100644 --- 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):