From 39d03fdcbfecdf6fedf73cda4db672064d21d90d Mon Sep 17 00:00:00 2001
From: suweiyue <suweiyue@baidu.com>
Date: Thu, 7 May 2020 20:25:03 +0800
Subject: [PATCH] fix erniesage graph_wrapper bug

---
 examples/erniesage/README.md      | 4 ++--
 examples/erniesage/local_run.sh   | 1 -
 examples/erniesage/models/base.py | 2 +-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/examples/erniesage/README.md b/examples/erniesage/README.md
index c4ac17a..cb375ff 100644
--- a/examples/erniesage/README.md
+++ b/examples/erniesage/README.md
@@ -39,10 +39,10 @@
 
 ```sh
 # 分布式GPU模式或单机模式ERNIESage
-sh local_run.sh config/enriesage_v1_gpu.yaml
+sh local_run.sh config/erniesage_v2_gpu.yaml
 
 # 分布式CPU模式训练ERNIESage
-sh local_run.sh config/enriesage_v1_cpu.yaml
+sh local_run.sh config/erniesage_v2_cpu.yaml
 ```
 
 ## Hyperparamters
diff --git a/examples/erniesage/local_run.sh b/examples/erniesage/local_run.sh
index 7b76d1c..d76d18c 100644
--- a/examples/erniesage/local_run.sh
+++ b/examples/erniesage/local_run.sh
@@ -57,7 +57,6 @@ collective_local_train(){
 }
 
 eval $(parse_yaml $config)
-unalias python
 
 python3 ./preprocessing/dump_graph.py -i $input_data -o $graph_path --encoding $encoding \
     -l $max_seqlen --vocab_file $ernie_vocab_file
diff --git a/examples/erniesage/models/base.py b/examples/erniesage/models/base.py
index c910c71..7df76ff 100644
--- a/examples/erniesage/models/base.py
+++ b/examples/erniesage/models/base.py
@@ -40,7 +40,7 @@ class BaseGraphWrapperBuilder(object):
             # all graph have same node_feat_info
             graph_wrappers.append(
                 pgl.graph_wrapper.GraphWrapper(
-                    "layer_%s" % i, place, node_feat=self.node_feature_info, edge_feat=self.edge_feature_info))
+                    "layer_%s" % i, node_feat=self.node_feature_info, edge_feat=self.edge_feature_info))
         return graph_wrappers
 
 
-- 
GitLab