zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update train.py remove save_json final_epoch

This commit is contained in:
Glenn Jocher 2020-08-09 21:24:40 -07:00
parent 41523e2c91
commit 8e5c66579b
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -306,7 +306,6 @@ def train(hyp, opt, device, tb_writer=None):
results, maps, times = test.test(opt.data, results, maps, times = test.test(opt.data,
batch_size=total_batch_size, batch_size=total_batch_size,
imgsz=imgsz_test, imgsz=imgsz_test,
save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'),
model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema, model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema,
single_cls=opt.single_cls, single_cls=opt.single_cls,
dataloader=testloader, dataloader=testloader,