未验证 提交 83585b55 编写于 作者: W whs 提交者: GitHub

Remove dependency on compat of PaddlePaddle (#1496)

上级 460b5c63
...@@ -28,7 +28,6 @@ import tarfile ...@@ -28,7 +28,6 @@ import tarfile
import zipfile import zipfile
import logging import logging
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.compat as cpt
from paddle.fluid import core from paddle.fluid import core
from paddle.fluid.framework import Program from paddle.fluid.framework import Program
...@@ -128,7 +127,7 @@ def load_persistable_nodes(executor, dirname, graph): ...@@ -128,7 +127,7 @@ def load_persistable_nodes(executor, dirname, graph):
persistable_nodes = [] persistable_nodes = []
all_persistable_nodes = graph.all_persistable_nodes() all_persistable_nodes = graph.all_persistable_nodes()
for node in all_persistable_nodes: for node in all_persistable_nodes:
name = cpt.to_text(node.name()) name = node.name()
if name not in persistable_node_names: if name not in persistable_node_names:
persistable_node_names.add(name) persistable_node_names.add(name)
persistable_nodes.append(node) persistable_nodes.append(node)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册