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('训练完成')