未验证 提交 c052dc67 编写于 作者: K KP 提交者: GitHub

Fix ernie_gen training error in windows device (#1540)

上级 ffd3cc34
......@@ -54,7 +54,8 @@ def convert_example(tokenizer,
else:
tgt_labels = tgt_ids
return (src_ids, src_pids, src_sids, tgt_ids, tgt_pids, tgt_sids, attn_ids, tgt_labels)
return [np.asarray(item, dtype=np.int64) for item \
in [src_ids, src_pids, src_sids, tgt_ids, tgt_pids, tgt_sids, attn_ids, tgt_labels]]
return warpper
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册