Skip to content
Snippets Groups Projects
Commit 1e6cc315 authored by anzhengqi's avatar anzhengqi
Browse files

modify yolov3_darknet53 network log info

parent ab85af4f
No related branches found
No related tags found
No related merge requests found
......@@ -280,12 +280,12 @@ def run_train():
if i % config.log_interval == 0:
time_used = time.time() - t_end
epoch = int(i / config.steps_per_epoch)
per_step_time = time_used/config.log_interval
per_step_time = time_used / config.log_interval * 1000
fps = config.per_batch_size * (i - old_progress) * config.group_size / time_used
if config.rank == 0:
config.logger.info(
'epoch[{}], iter[{}], {}, {:.2f} imgs/sec, lr:{},'
' per_step_time:{}'.format(epoch, i, loss_meter, fps, lr[i], per_step_time))
' per step time:{}ms'.format(epoch, i, loss_meter, fps, lr[i], per_step_time))
t_end = time.time()
loss_meter.reset()
old_progress = i
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment