关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

New Pull Request

#156 更新 'train.py'

Merged

WCX merged 1 commits from wcx-patch-4 into main 1 year ago

1 changed files with 1 additions and 1 deletions

Split View

							
								+ 1
								
								- 1
							
train.pyView File
								
			@@ -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