From 5081e5b2a3ec1f96a77df881d8946d750332705c Mon Sep 17 00:00:00 2001 From: OutisLi Date: Wed, 22 Apr 2026 09:17:44 +0800 Subject: [PATCH] fix: remove meaningless error raising --- deepmd/pt/train/training.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py index da08f62e2b..8ae7ccfba3 100644 --- a/deepmd/pt/train/training.py +++ b/deepmd/pt/train/training.py @@ -183,9 +183,7 @@ def __init__( f"training.zero_stage must be 0, 1, 2, or 3, got {self.zero_stage}" ) if self.zero_stage > 0 and not self.is_distributed: - raise ValueError( - "training.zero_stage requires distributed launch via torchrun." - ) + self.zero_stage = 0 if self.zero_stage > 0 and self.change_bias_after_training: raise ValueError( "training.zero_stage does not support change_bias_after_training."