feat(train): 添加模型输入功能- 在训练过程中增加了用户输入模型名称的功能

- 打印出用户输入的模型名称

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'