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: