From 97c3fcf4af4cbc839de802c1017dfc211d622f08 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
<66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 27 Nov 2024 04:49:00 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
---
deepmd/pt/train/training.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
index 16c7810727..af6e48191d 100644
--- a/deepmd/pt/train/training.py
+++ b/deepmd/pt/train/training.py
@@ -764,7 +764,7 @@ def fake_model():
if self.display_in_training and (
display_step_id % self.disp_freq == 0 or display_step_id == 1
):
- self.wrapper.eval() # Will set to train mode before fininshing validation
+ self.wrapper.eval() # Will set to train mode before fininshing validation
def log_loss_train(_loss, _more_loss, _task_key="Default"):
results = {}
@@ -925,6 +925,7 @@ def log_loss_valid(_task_key="Default"):
writer.add_scalar(
f"{task_key}/{item}", more_loss[item], display_step_id
)
+
self.wrapper.train()
self.t0 = time.time()
self.total_train_time = 0.0