fix bug for kunlun ()

pull/524/head
QingshuChen 2020-12-28 13:43:05 +08:00 committed by GitHub
parent 11ef05ca32
commit 918e68a934
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
tools/static

View File

@ -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: