diff --git a/train_yolov8.py b/train_yolov8.py
index 75561fa..eb45630 100644
--- a/train_yolov8.py
+++ b/train_yolov8.py
@@ -7,7 +7,7 @@ from ultralytics import YOLO
 def parse_args():
     parser = ArgumentParser(description="Train YOLOv8 on this dataset.")
     parser.add_argument("--model", default="yolov8n.pt", help="Model or checkpoint path.")
-    parser.add_argument("--epochs", type=int, default=100)
+    parser.add_argument("--epochs", type=int, default=1)
     parser.add_argument("--imgsz", type=int, default=640)
     parser.add_argument("--batch", type=int, default=8, help="Use -1 for automatic batch size.")
     parser.add_argument("--device", default="0", help="CUDA device such as 0, or cpu.")