From 716a15b4340f0b9f7da0200a00456097bb76cfba Mon Sep 17 00:00:00 2001 From: OutisLi Date: Wed, 22 Apr 2026 09:07:09 +0800 Subject: [PATCH 1/2] fix: typo --- deepmd/pt/train/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py index da08f62e2b..96a4a578b8 100644 --- a/deepmd/pt/train/training.py +++ b/deepmd/pt/train/training.py @@ -164,7 +164,7 @@ def __init__( # Iteration config self.num_steps = training_params.get("numb_steps") - self.num_epoch = training_params.get("num_epoch") + self.num_epoch = training_params.get("numb_epoch") self.num_epoch_dict = training_params.get("num_epoch_dict") self.disp_file = training_params.get("disp_file", "lcurve.out") self.disp_freq = training_params.get("disp_freq", 1000) From 86a93af05581132aafdab19472ccb3488417796b Mon Sep 17 00:00:00 2001 From: OutisLi Date: Wed, 22 Apr 2026 13:46:59 +0800 Subject: [PATCH 2/2] fix --- deepmd/pd/train/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pd/train/training.py b/deepmd/pd/train/training.py index 3c848e68c1..7d91218468 100644 --- a/deepmd/pd/train/training.py +++ b/deepmd/pd/train/training.py @@ -141,7 +141,7 @@ def __init__( # Iteration config self.num_steps = training_params.get("numb_steps") - self.num_epoch = training_params.get("num_epoch") + self.num_epoch = training_params.get("numb_epoch") self.num_epoch_dict = training_params.get("num_epoch_dict") self.acc_freq: int = training_params.get( "acc_freq", 1