fix bug for kunlun (#518)
parent
11ef05ca32
commit
918e68a934
tools/static
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue