This commit is contained in:
mshe 2025-09-27 20:41:59 +08:00
parent d892c72411
commit 061e74011d
1 changed files with 1 additions and 1 deletions

View File

@ -11,7 +11,7 @@ model = YOLO('yolov8n-pose.pt')
# 3080优化配置 # 3080优化配置
model.train( model.train(
data='./dataset/train.yaml', data='./dataset_tiger/train.yaml',
epochs=300, epochs=300,
imgsz=640, # 3080可以处理原尺寸 imgsz=640, # 3080可以处理原尺寸
batch=8, # 根据12GB内存调整 batch=8, # 根据12GB内存调整