From e7be3bd99c5d31a65066200fae25bf4adf6c1ab1 Mon Sep 17 00:00:00 2001 From: Zhen Wang Date: Tue, 2 Apr 2019 15:39:09 +0800 Subject: [PATCH] add compat for string. --- fluid/PaddleSlim/utility.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fluid/PaddleSlim/utility.py b/fluid/PaddleSlim/utility.py index c2a8cd53..b0848135 100644 --- a/fluid/PaddleSlim/utility.py +++ b/fluid/PaddleSlim/utility.py @@ -21,6 +21,7 @@ import os import numpy as np import six import paddle.fluid as fluid +import paddle.compat as cpt from paddle.fluid import core from paddle.fluid.framework import Program @@ -79,7 +80,7 @@ def save_persistable_nodes(executor, dirname, graph): persistable_nodes = [] all_persistable_nodes = graph.all_persistable_nodes() for node in all_persistable_nodes: - name = node.name() + name = cpt.to_text(node.name()) if name not in persistable_node_names: persistable_node_names.add(name) persistable_nodes.append(node) @@ -114,7 +115,7 @@ def load_persistable_nodes(executor, dirname, graph): persistable_nodes = [] all_persistable_nodes = graph.all_persistable_nodes() for node in all_persistable_nodes: - name = node.name() + name = cpt.to_text(node.name()) if name not in persistable_node_names: persistable_node_names.add(name) persistable_nodes.append(node) -- GitLab