From ad0b45e198db6438472c5e2d631e85524b065336 Mon Sep 17 00:00:00 2001
From: zjut <ZJUTB519@zjut.edu.cn>
Date: Sun, 17 Nov 2024 16:15:26 +0800
Subject: [PATCH] =?UTF-8?q?feat(train):=20=E6=B7=BB=E5=8A=A0=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E8=BE=93=E5=85=A5=E5=8A=9F=E8=83=BD-=20=E5=9C=A8?=
 =?UTF-8?q?=E8=AE=AD=E7=BB=83=E8=BF=87=E7=A8=8B=E4=B8=AD=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E4=BA=86=E7=94=A8=E6=88=B7=E8=BE=93=E5=85=A5=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=90=8D=E7=A7=B0=E7=9A=84=E5=8A=9F=E8=83=BD=20-=20=E6=89=93?=
 =?UTF-8?q?=E5=8D=B0=E5=87=BA=E7=94=A8=E6=88=B7=E8=BE=93=E5=85=A5=E7=9A=84?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=90=8D=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/train.py b/train.py
index bf43c31..583fdb2 100644
--- a/train.py
+++ b/train.py
@@ -80,6 +80,9 @@ print(f"Clip gradient norm value: {clip_grad_norm_value}")
 print(f"Optimization step: {optim_step}")
 print(f"Optimization gamma: {optim_gamma}")
 
+# 控制台输入
+model_str = input("Model: ")
+print(f"Model: {model_str}")
 
 # Model
 device = 'cuda' if torch.cuda.is_available() else 'cpu'