Skip to content

GitLab

Commit d923cf4c authored by ThangVu's avatar ThangVu
Browse files

add eta to log

parent a097c65f
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
mmcv/runner/hooks/logger/text.py
View file @ d923cf4c
from .base import LoggerHook from .base import LoggerHook
import datetime
class TextLoggerHook(LoggerHook): class TextLoggerHook(LoggerHook):
...@@ -14,6 +15,12 @@ class TextLoggerHook(LoggerHook): ...@@ -14,6 +15,12 @@ class TextLoggerHook(LoggerHook):
log_str = 'Epoch({}) [{}][{}]\t'.format(runner.mode, runner.epoch, log_str = 'Epoch({}) [{}][{}]\t'.format(runner.mode, runner.epoch,
runner.inner_iter + 1) runner.inner_iter + 1)
if 'time' in runner.log_buffer.output: if 'time' in runner.log_buffer.output:
tot_time_sec = runner.log_buffer.output['time'] + \
runner.log_buffer.output['data_time']
eta_sec = tot_time_sec*(len(runner.data_loader)*runner.max_epochs - \
runner._iter)
eta_str = str(datetime.timedelta(seconds=int(eta_sec)))
log_str += ('eta: {}, '.format(eta_str))
log_str += ( log_str += (
'time: {log[time]:.3f}, data_time: {log[data_time]:.3f}, '. 'time: {log[time]:.3f}, data_time: {log[data_time]:.3f}, '.
format(log=runner.log_buffer.output)) format(log=runner.log_buffer.output))
......
Markdown is supported
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