simple setup

.gitignore

@@ -37,3 +37,5 @@ wheels/
 # OS
 .DS_Store
 Thumbs.db
+
+datasets

train_script.py

@@ -0,0 +1,18 @@
+from ultralytics import YOLO
+
+def main():
+    model = YOLO('yolov8n.pt') 
+
+    model.train(
+        data='datasets/data.yaml', 
+        epochs=50, 
+        imgsz=640, 
+        device='cpu',
+        # --- RESOURCE LIMITS ---
+        batch=4,          # Lower = less RAM used. Start with 4. 
+        workers=1,        # 1 or 2. Prevents the CPU from maxing all cores for data loading.
+        cache=False       # Don't store the whole dataset in RAM.
+    )
+
+if __name__ == '__main__':
+    main()