Skip to content

Commit 5665c0f

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 6ad353f commit 5665c0f

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

deepmd/pt/train/training.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -936,8 +936,10 @@ def log_loss_valid(_task_key="Default"):
936936
)
937937
)
938938
if (
939-
(self.num_steps - self.start_step) <= 2 * self.disp_freq # not enough steps
940-
or (_step_id - self.start_step) >= self.disp_freq # skip first disp_freq steps
939+
(self.num_steps - self.start_step)
940+
<= 2 * self.disp_freq # not enough steps
941+
or (_step_id - self.start_step)
942+
>= self.disp_freq # skip first disp_freq steps
941943
):
942944
self.total_train_time += train_time
943945

0 commit comments

Comments
 (0)