From efdb4be1cded334001b214fa3a9460260663f460 Mon Sep 17 00:00:00 2001 From: WCX <1084321207@qq.com> Date: Tue, 28 Mar 2023 10:49:45 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'train.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 02928b0..74bb07c 100644 --- a/train.py +++ b/train.py @@ -15,7 +15,7 @@ import torch#torch库 拥有最多的基础模型 import os import sys#系统 import numpy as np #numpy库 -import yaml +import yaml#输入参数 from tabulate import tabulate from torch.optim.lr_scheduler import StepLR, ExponentialLR from warmup_scheduler import GradualWarmupScheduler -- 2.34.1