Remove is_coco argument from test() (#3553)
This commit is contained in:
1 parent
958ab92dc1
commit
63157d214d
2 files changed
+3
-6
No files matched your search
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user