未验证 提交 9957111d 编写于 作者: M Meiyim 提交者: GitHub

Merge pull request #289 from P01son6415/fix

Fix TypeError on 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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册