From f2bd169955812f21ef2fa3b16b5fd8cd74dc5903 Mon Sep 17 00:00:00 2001
From: songbingle <1696321477@qq.com>
Date: Fri, 6 Jun 2025 18:05:09 +0800
Subject: [PATCH] =?UTF-8?q?[MF]=E4=BF=AE=E6=94=B9=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/data.yaml | 2 +-
 src/train.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/data.yaml b/src/data.yaml
index 67aa956..6fcfe20 100644
--- a/src/data.yaml
+++ b/src/data.yaml
@@ -1,4 +1,4 @@
-path: C:\workspace\le-yolo\data
+path: /home/le/le-yolo/data
 train: images/train
 val: images/val
 test: images/test
diff --git a/src/train.py b/src/train.py
index b8fb234..921eb78 100644
--- a/src/train.py
+++ b/src/train.py
@@ -3,7 +3,7 @@ import time
 start_time = time.time()
 model = YOLO("yolov8n.pt")
 # model = YOLO(r"C:\workspace\le-yolo\runs\detect\train40\weights\last.pt")
-model.train(data="data.yaml", epochs=100,  batch=8, device='cpu',  imgsz=640, augment = True)
+model.train(data="data.yaml", epochs=100,  batch=8, device=0,  imgsz=640, augment = True)
 # model.train(data="data.yaml", epochs=100,  batch=8, device='cpu',  imgsz=640, augment = True, lr = 0.001,wight_decay  = 0.0005 )
 model.val()
 print('训练完成')