Update deepmd/pt/train/training.py

OutisLi · gemini-code-assist[bot] · OutisLi · commit aefa5e005706 · 2026-02-14T12:32:23.000+08:00
Co-authored-by: gemini-code-assist[bot] &lt;176961590+gemini-code-assist[bot]@users.noreply.github.com&gt;
Signed-off-by: OutisLi &lt;137472077+OutisLi@users.noreply.github.com&gt;
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
@@ -728,7 +728,7 @@ def single_model_finetune(
                 self.wrapper = DDP(
                     self.wrapper,
                     device_ids=[LOCAL_RANK],
-                    find_unused_parameters=self.multi_task,
+                    find_unused_parameters=True,
                     output_device=LOCAL_RANK,
                 )
 

Original file line number	Diff line number	Diff line change
`@@ -728,7 +728,7 @@ def single_model_finetune(`
`728`	`728`	`self.wrapper = DDP(`
`729`	`729`	`self.wrapper,`
`730`	`730`	`device_ids=[LOCAL_RANK],`
`731`		`- find_unused_parameters=self.multi_task,`
	`731`	`+ find_unused_parameters=True,`
`732`	`732`	`output_device=LOCAL_RANK,`
`733`	`733`	`)`
`734`	`734`