diff --git a/train_yolov8.py b/train_yolov8.py
index 9461359..2caf02a 100644
--- a/train_yolov8.py
+++ b/train_yolov8.py
@@ -9,7 +9,7 @@ def parse_args():
     parser.add_argument("--model", default="yolov8n.pt", help="Model or checkpoint path.")
     parser.add_argument("--epochs", type=int, default=1)
     parser.add_argument("--imgsz", type=int, default=1024)
-    parser.add_argument("--batch", type=int, default=16, help="Use -1 for automatic batch size.")
+    parser.add_argument("--batch", type=int, default=9, help="Use -1 for automatic batch size.")
     parser.add_argument("--device", default="0", help="CUDA device such as 0, or cpu.")
     parser.add_argument("--workers", type=int, default=4)
     parser.add_argument("--patience", type=int, default=30)