未验证 提交 918e68a9 编写于 作者: Q QingshuChen 提交者: GitHub

fix bug for kunlun (#518)

上级 11ef05ca
......@@ -124,7 +124,7 @@ def main(args):
# load pretrained models or checkpoints
init_model(config, train_prog, exe)
if not config.get("is_distributed", True):
if not config.get("is_distributed", True) and not use_xpu:
compiled_train_prog = program.compile(
config, train_prog, loss_name=train_fetchs["loss"][0].name)
else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册