train fixes merge

2026-04-22 08:46:36 +00:00 · 2025-05-22 16:56:09 +03:00
parent ce29d7fc58
commit 5b90320e27
3 changed files with 12 additions and 4 deletions
@@ -10,4 +10,9 @@ models/
 *.mp4
 venv
 *.engine
+<<<<<<< HEAD
 *.log
+=======
+*.pyc
+
+>>>>>>> 6146f2b (train fixes)
@@ -1,5 +1,8 @@
 from augmentation import Augmentator
-from train import train_dataset, convert2rknn
+from train import train_dataset, convert2rknn, resume_training

-Augmentator().augment_annotations()
-train_dataset(from_scratch=True)
+# Augmentator().augment_annotations()
+#train_dataset(from_scratch=True)
+
+
+resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
@@ -177,7 +177,7 @@ def resume_training(last_pt_path):
    results = model.train(data=yaml,
                          resume=True,
                          epochs=120,
-                          batch=12,
+                          batch=11,
                          imgsz=1280,
                          save_period=1,
                          workers=24)