diff --git a/examples/erniesage/README.md b/examples/erniesage/README.md
index c4ac17ae5e4a345fb8d1681d61d7bcfe866513f4..cb375fff3c4129953b2c99a30cb38669ea25c9d8 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 7b76d1c8288472d5c84abfd8585032c71ace8dab..d76d18c14057ddcb06b7d8b5aa5edac3357587c5 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 c910c714db4269bc9b3bac391e2adf2699f476c6..7df76fff9917d6ca22c0e8a98e5907209e954f88 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