diff --git a/.gitignore b/.gitignore
index dbf3c2f..5368cd9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,4 +10,9 @@ models/
 *.mp4
 venv
 *.engine
+<<<<<<< HEAD
 *.log
+=======
+*.pyc
+
+>>>>>>> 6146f2b (train fixes)
diff --git a/preprocess-train.py b/preprocess-train.py
index 610773a..9216a9e 100644
--- a/preprocess-train.py
+++ b/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()
-train_dataset(from_scratch=True)
\ No newline at end of file
+# Augmentator().augment_annotations()
+#train_dataset(from_scratch=True)
+
+
+resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
diff --git a/train.py b/train.py
index dba64cc..346580a 100644
--- a/train.py
+++ b/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)