From fb38cd8bba3763bd05dedf22a282bf59e91c6e8d Mon Sep 17 00:00:00 2001
From: mshe <666666666@666666666.666666666>
Date: Sat, 27 Sep 2025 19:47:49 +0800
Subject: [PATCH] bak

---
 14.训练模型-姿态估计-windows.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/14.训练模型-姿态估计-windows.py b/14.训练模型-姿态估计-windows.py
index b3cf06c..a9685b2 100644
--- a/14.训练模型-姿态估计-windows.py
+++ b/14.训练模型-姿态估计-windows.py
@@ -7,14 +7,14 @@ print(f"GPU数量: {torch.cuda.device_count()}")
 print(f"当前GPU: {torch.cuda.get_device_name(0)}")
 print(f"GPU内存: {torch.cuda.get_device_properties(0).total_memory / 1024 ** 3:.1f} GB")
 
-model = YOLO('models/yolov8n-pose.pt')
+model = YOLO('yolov8n-pose.pt')
 
 # 3080优化配置
 model.train(
     data='./dataset1/train.yaml',
     epochs=300,
     imgsz=640,  # 3080可以处理原尺寸
-    batch=32,  # 根据12GB内存调整
+    batch=8,  # 根据12GB内存调整
     device=0,  # 使用GPU 0
     workers=8,  # 充分利用CPU核心
     patience=50,