Update train.py

nikitabelooussovbtis · Jun 23, 2022 · ee1e6dd · ee1e6dd
1 parent b2f5e19
commit ee1e6dd
Showing 1 changed file with 1 addition and 1 deletion.
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=False,
+                                              augment=True,
                                               cache=None if opt.cache == 'val' else opt.cache,
                                               rect=opt.rect,
                                               rank=LOCAL_RANK,