sun15684712291/aicheckv2
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化训练过程

Browse Source
This commit is contained in:
sunyg
2025-03-20 16:41:51 +08:00
parent 358bb40a2a
commit bba39adcfc
7 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
requirements_gpu.txt
View File

@ -31,11 +31,11 @@ PyYAML>=5.3.1
requests>=2.32.2
scipy==1.13.1
thop>=0.1.1 # FLOPs computation
torch==2.3.1+cu121 --extra-index-url https://mirrors.aliyun.com/pytorch-wheels/cu121/
torchvision==0.18.1+cu121 --extra-index-url https://mirrors.aliyun.com/pytorch-wheels/cu121/
# torch==2.6.0+cu124 # 本地安装
# torchvision==0.21.0+cu124 # 本地安装
tqdm>=4.66.3
ultralytics==8.3.75 # https://ultralytics.com
# Plotting --------------------------------------------------------------------
pandas==2.2.3
seaborn==0.11.0 # 对应的这个依赖也需要降
seaborn==0.11.2 # 对应的这个依赖也需要降