bak
This commit is contained in:
parent
061e74011d
commit
f712c858c9
|
|
@ -11,7 +11,7 @@ model = YOLO('yolov8n-pose.pt')
|
|||
|
||||
# 3080优化配置
|
||||
model.train(
|
||||
data='./dataset_tiger/train.yaml',
|
||||
data='./dataset/train.yaml',
|
||||
epochs=300,
|
||||
imgsz=640, # 3080可以处理原尺寸
|
||||
batch=8, # 根据12GB内存调整
|
||||
|
|
|
|||
Loading…
Reference in New Issue