From 2219c0a3035e8f7baab5563a24306ad5b4cc53ce Mon Sep 17 00:00:00 2001
From: lee <770918727@qq.com>
Date: Wed, 2 Jul 2025 14:53:37 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train_compare.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/train_compare.py b/train_compare.py
index 3cd238b..5592155 100644
--- a/train_compare.py
+++ b/train_compare.py
@@ -11,7 +11,7 @@ from tools.dataset import load_data
 import matplotlib.pyplot as plt
 from configs import trainer_tools
 import yaml
-
+from datetime import datetime
 with open('configs/scatter.yml', 'r') as f:
     conf = yaml.load(f, Loader=yaml.FullLoader)
 
@@ -125,8 +125,13 @@ if __name__ == '__main__':
 
         scheduler.step()
         current_lr = optimizer.param_groups[0]['lr']
-        log_info = ("Epoch {}/{},  train_loss: {}, val_loss: {} lr:{}"
-                    .format(e, conf['training']['epochs'], train_lossAvg, val_lossAvg, current_lr))
+        log_info = ("[{:%Y-%m-%d %H:%M:%S}] Epoch {}/{},  train_loss: {}, val_loss: {} lr:{}"
+                    .format(datetime.now(),
+                            e,
+                            conf['training']['epochs'],
+                            train_lossAvg,
+                            val_lossAvg,
+                            current_lr))
         print(log_info)
         # 写入日志文件
         with open(osp.join(conf['logging']['logging_dir']), 'a') as f: