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