fix str

apache · Jun 26, 2020 · b815c4e · b815c4e
1 parent 318d268
commit b815c4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/backend/graph_runtime_codegen.py b/python/tvm/relay/backend/graph_runtime_codegen.py
@@ -85,7 +85,7 @@ def codegen(self, func):
         param_names = self._list_params_name()
         params = {}
         for key in param_names:
-            arr = self._get_param_by_name(key)
+            arr = self._get_param_by_name(str(key))
             param = empty(arr.shape, dtype=arr.dtype, ctx=arr.ctx)
             arr.copyto(param)
             params[key] = param