diff --git a/mmcv/runner/hooks/logger/text.py b/mmcv/runner/hooks/logger/text.py index 06ac3475e..8711cde1c 100644 --- a/mmcv/runner/hooks/logger/text.py +++ b/mmcv/runner/hooks/logger/text.py @@ -1,4 +1,5 @@ from .base import LoggerHook +import datetime class TextLoggerHook(LoggerHook): @@ -14,6 +15,12 @@ class TextLoggerHook(LoggerHook): log_str = 'Epoch({}) [{}][{}]\t'.format(runner.mode, runner.epoch, runner.inner_iter + 1) 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 += ( 'time: {log[time]:.3f}, data_time: {log[data_time]:.3f}, '. format(log=runner.log_buffer.output))