Commit
·
8e5c665
1
Parent(s):
41523e2
update train.py remove save_json final_epoch
Browse files
train.py
CHANGED
@@ -306,7 +306,6 @@ def train(hyp, opt, device, tb_writer=None):
|
|
306 |
results, maps, times = test.test(opt.data,
|
307 |
batch_size=total_batch_size,
|
308 |
imgsz=imgsz_test,
|
309 |
-
save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'),
|
310 |
model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema,
|
311 |
single_cls=opt.single_cls,
|
312 |
dataloader=testloader,
|
|
|
306 |
results, maps, times = test.test(opt.data,
|
307 |
batch_size=total_batch_size,
|
308 |
imgsz=imgsz_test,
|
|
|
309 |
model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema,
|
310 |
single_cls=opt.single_cls,
|
311 |
dataloader=testloader,
|