update fast mode

2020-06-16 00:11:29 -07:00
1 parent db2c3acd3a
commit 1c0b6236e3
@@ -303,7 +303,7 @@ def train(hyp):
                                             model=ema.ema,
                                             single_cls=opt.single_cls,
                                             dataloader=testloader,
-                                             fast=ni < n_burn)
+                                             fast=epoch > epochs / 2)

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