updates

5 years ago · 11121e39ed
parent dbdee3a4a3
commit 11121e39ed
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -299,7 +299,7 @@ def train(hyp):
                                             model=ema.ema,
                                             single_cls=opt.single_cls,
                                             dataloader=testloader,
-                                             fast=ni > n_burn)
+                                             fast=ni < n_burn)

        # Write
        with open(results_file, 'a') as f: