diff --git a/train.py b/train.py index 27228cd6f..6552a6e6b 100644 --- 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: