Merge pull request #289 from P01son6415/fix

Fix TypeError on ernie_encoder.py
PaddlePaddle · Aug 27, 2019 · 9957111 · 9957111
2 parents 962d926 + 7d32579
commit 9957111
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ernie_encoder.py b/ernie_encoder.py
@@ -168,10 +168,10 @@ def main(args):
     total_top_layer_emb = np.concatenate(total_top_layer_emb)
 
     with open(os.path.join(args.output_dir, "cls_emb.npy"),
-              "w") as cls_emb_file:
+              "wb") as cls_emb_file:
         np.save(cls_emb_file, total_cls_emb)
     with open(os.path.join(args.output_dir, "top_layer_emb.npy"),
-              "w") as top_layer_emb_file:
+              "wb") as top_layer_emb_file:
         np.save(top_layer_emb_file, total_top_layer_emb)