提交 7d325793 编写于 作者: P P01son6415

modify mode of open() to 'wb'

上级 815159f4
......@@ -170,10 +170,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.
先完成此消息的编辑!
想要评论请 注册