train fixes merge

.gitignore

@@ -10,4 +10,9 @@ models/
 *.mp4
 venv
 *.engine
+<<<<<<< HEAD
 *.log
+=======
+*.pyc
+
+>>>>>>> 6146f2b (train fixes)

preprocess-train.py

@@ -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()
+# Augmentator().augment_annotations()
-train_dataset(from_scratch=True)
+#train_dataset(from_scratch=True)
+
+
+resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')

train.py

@@ -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)