Update train.py

nikitabelooussovbtis · Jun 23, 2022 · b2f5e19 · b2f5e19
1 parent 54e7f34
commit b2f5e19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -309,7 +309,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                                                   gs,
                                                   single_cls,
                                                   hyp=hyp,
-                                                  augment=False,
+                                                  augment=True,
                                                   cache=None if opt.cache == 'val' else opt.cache,
                                                   rect=opt.rect,
                                                   rank=LOCAL_RANK,