Update train.py

nikitabelooussovbtis · Jun 22, 2022 · 54e7f34 · 54e7f34
1 parent 1194650
commit 54e7f34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -224,7 +224,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                                               gs,
                                               single_cls,
                                               hyp=hyp,
-                                              augment=True,
+                                              augment=False,
                                               cache=None if opt.cache == 'val' else opt.cache,
                                               rect=opt.rect,
                                               rank=LOCAL_RANK,
@@ -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=True,
+                                                  augment=False,
                                                   cache=None if opt.cache == 'val' else opt.cache,
                                                   rect=opt.rect,
                                                   rank=LOCAL_RANK,