Merge pull request #67 from WeiyueSu/erniesage

Erniesage

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

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

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