Integer printout (#2450)
* Integer printout * test.py 'Labels' * Update train.py
This commit is contained in:
1 parent
886f1c03d8
commit
f01f3223d5
3 files changed
+4
-4
No files matched your search
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user