Remove is_coco argument from test() (#3553)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-09 15:09:51 +02:00
1 parent 958ab92dc1
commit 63157d214d
2 files changed
+3 -6

No files matched your search

+2 -4
View File
@@ -365,8 +365,7 @@ def train(hyp, opt, device, tb_writer=None):
verbose=nc < 50 and final_epoch,
plots=plots and final_epoch,
wandb_logger=wandb_logger,
compute_loss=compute_loss,
is_coco=is_coco)
compute_loss=compute_loss)
# Write
with open(results_file, 'a') as f:
@@ -434,8 +433,7 @@ def train(hyp, opt, device, tb_writer=None):
dataloader=testloader,
save_dir=save_dir,
save_json=True,
plots=False,
is_coco=is_coco)
plots=False)
# Strip optimizers
for f in last, best: