You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ar...@apache.org on 2021/05/04 22:15:16 UTC

[tvm] branch main updated: [FIX] Fix deploy_sparse tutorial (#7939)

This is an automated email from the ASF dual-hosted git repository.

areusch pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new dea4190  [FIX] Fix deploy_sparse tutorial (#7939)
dea4190 is described below

commit dea4190234c7d8d3a71649abde08afce990bdb74
Author: Tristan Konolige <tr...@gmail.com>
AuthorDate: Tue May 4 15:14:57 2021 -0700

    [FIX] Fix deploy_sparse tutorial (#7939)
---
 tutorials/frontend/deploy_sparse.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tutorials/frontend/deploy_sparse.py b/tutorials/frontend/deploy_sparse.py
index 1fcb1b3..92f4511 100644
--- a/tutorials/frontend/deploy_sparse.py
+++ b/tutorials/frontend/deploy_sparse.py
@@ -198,7 +198,7 @@ def import_graphdef(
             with open(os.path.join(abs_path, relay_params), "wb") as fo:
                 fo.write(runtime.save_param_dict(params))
 
-    return mod, params, shape_dict
+    return mod, dict(params.items()), shape_dict
 
 
 ###############################################################################