Skip to content

Commit fb60645

Browse files
[AutoParallel] Fix ernie auto_trainer error (PaddlePaddle#9753)
1 parent 1d74d62 commit fb60645

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

paddlenlp/trainer/auto_trainer.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,15 @@ def loss_func(loss, outputs):
105105
model = kwargs["model"]
106106
for param in model.parameters():
107107
if not param._is_initialized():
108-
param.initialize()
108+
try:
109+
param.initialize()
110+
except Exception as e:
111+
# NOTE(zhangwl):maybe param is not initialized and param init_func is set in later.user need set_init_func before auto_trainer
112+
logger.warning(
113+
f"AutoTrainer requires all parameters to be initialized when auto_trainer init, but failed to initialize parameter {param.name} {param}.\n"
114+
+ "Please check param init func.\n"
115+
+ f"The original exception message is:\n{str(e)}"
116+
)
109117
kwargs["model"] = model
110118

111119
super().__init__(*args, **kwargs)

0 commit comments

Comments
 (0)