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