File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed
Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments