Integer printout (#2450)

* Integer printout

* test.py 'Labels'

* Update train.py
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-12 13:35:44 -08:00
1 parent 886f1c03d8
commit f01f3223d5
3 files changed
+4 -4

No files matched your search

+1 -1
View File
@@ -264,7 +264,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
if rank != -1:
dataloader.sampler.set_epoch(epoch)
pbar = enumerate(dataloader)
logger.info(('\n' + '%10s' * 8) % ('Epoch', 'gpu_mem', 'box', 'obj', 'cls', 'total', 'targets', 'img_size'))
logger.info(('\n' + '%10s' * 8) % ('Epoch', 'gpu_mem', 'box', 'obj', 'cls', 'total', 'labels', 'img_size'))
if rank in [-1, 0]:
pbar = tqdm(pbar, total=nb) # progress bar
optimizer.zero_grad()