diff --git a/vissl/hooks/log_hooks.py b/vissl/hooks/log_hooks.py index a3f66f830..3245bfe5a 100644 --- a/vissl/hooks/log_hooks.py +++ b/vissl/hooks/log_hooks.py @@ -4,7 +4,6 @@ All the hooks involved in human-readable logging """ -import datetime import logging import time from typing import Optional @@ -197,12 +196,9 @@ def on_update(self, task: "tasks.ClassyTask") -> None: rolling_eta_secs = int( rolling_avg_time * (task.max_iteration - iteration) ) - rolling_eta_str = str( - datetime.timedelta(seconds=int(rolling_eta_secs)) - ) rolling_btime = int(1000.0 * rolling_avg_time) evt_stg.put_scalars( - rolling_btime=rolling_btime, rolling_eta=rolling_eta_str + rolling_btime=rolling_btime, rolling_eta=rolling_eta_secs ) for writer in task.event_storage_writers: writer.write()