Skip to content

Commit

Permalink
W&B DDP fix 2 (ultralytics#2587)
Browse files Browse the repository at this point in the history
Revert unintentional change to test batch sizes caused by PR ultralytics#2125
  • Loading branch information
glenn-jocher authored Mar 24, 2021
1 parent 2bcc89d commit 9f98201
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,7 +349,7 @@ def train(hyp, opt, device, tb_writer=None):
if not opt.notest or final_epoch: # Calculate mAP
wandb_logger.current_epoch = epoch + 1
results, maps, times = test.test(data_dict,
batch_size=total_batch_size,
batch_size=batch_size * 2,
imgsz=imgsz_test,
model=ema.ema,
single_cls=opt.single_cls,
Expand Down

0 comments on commit 9f98201

Please sign in to comment.