From 5ece085064e9ced2282d68980993d5317017aa1e Mon Sep 17 00:00:00 2001
From: ferdzo <andrej@ferdzo.xyz>
Date: Thu, 26 Jun 2025 18:41:51 +0200
Subject: [PATCH] Added small changes and migrated to Yolo v11

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 04093f2..6e1b3f9 100644
--- a/train.py
+++ b/train.py
@@ -21,6 +21,7 @@ train_params = {
     'exist_ok': True,
     'save_period': 2,
     'workers': 8,
+    'auto_augment': 'autoaugment',
 }
 
 model.train(data=data_path, **train_params)
\ No newline at end of file