Commit
•
11121e3
1
Parent(s):
dbdee3a
updates
Browse files
train.py
CHANGED
|
@@ -299,7 +299,7 @@ def train(hyp):
|
|
| 299 |
model=ema.ema,
|
| 300 |
single_cls=opt.single_cls,
|
| 301 |
dataloader=testloader,
|
| 302 |
-
fast=ni
|
| 303 |
|
| 304 |
# Write
|
| 305 |
with open(results_file, 'a') as f:
|
|
|
|
| 299 |
model=ema.ema,
|
| 300 |
single_cls=opt.single_cls,
|
| 301 |
dataloader=testloader,
|
| 302 |
+
fast=ni < n_burn)
|
| 303 |
|
| 304 |
# Write
|
| 305 |
with open(results_file, 'a') as f:
|