fix neg_type bug

examples/erniesage/config/erniesage_v2_cpu.yaml

@@ -31,7 +31,7 @@ final_fc: true
 final_l2_norm: true
 loss_type: "hinge"
 margin: 0.3
-neg_type: "random_neg"
+neg_type: "batch_neg"
 
 # infer config ------
 infer_model: "./output/last"

examples/erniesage/config/erniesage_v2_gpu.yaml

@@ -31,7 +31,7 @@ final_fc: true
 final_l2_norm: true
 loss_type: "hinge"
 margin: 0.3
-neg_type: "random_neg"
+neg_type: "batch_neg"
 
 # infer config ------
 infer_model: "./output/last"

examples/erniesage/dataset/graph_reader.py

@@ -73,7 +73,7 @@ class GraphGenerator(BaseDataGenerator):
         batch_src = np.array(batch_src, dtype="int64")
         batch_dst = np.array(batch_dst, dtype="int64")
 
-        if neg_type == "batch_neg":
+        if self.neg_type == "batch_neg":
             neg_shape = [1]
         else:
             neg_shape = batch_dst.shape