ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 05:18:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Black formatting

Browse Source
This commit is contained in:
allegroai 2024-06-16 23:15:26 +03:00
parent 9fff3bc03c
commit cf4134ee02
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
clearml/debugging/timer.py
View File

@ -6,23 +6,23 @@ import six
class Timer(object): class Timer(object):
"""A class implementing a simple timer, with a reset option """ """A class implementing a simple timer, with a reset option"""
def __init__(self): def __init__(self):
self._start_time = 0. self._start_time = 0.0
self._diff = 0. self._diff = 0.0
self._total_time = 0. self._total_time = 0.0
self._average_time = 0. self._average_time = 0.0
self._calls = 0 self._calls = 0
self.tic() self.tic()
def reset(self): def reset(self):
self._start_time = 0. self._start_time = 0.0
self._diff = 0. self._diff = 0.0
self.reset_average() self.reset_average()
def reset_average(self): def reset_average(self):
""" Reset average counters (does not change current timer) """ """Reset average counters (does not change current timer)"""
self._total_time = 0 self._total_time = 0
self._average_time = 0 self._average_time = 0
self._calls = 0 self._calls = 0
@ -54,7 +54,7 @@ class Timer(object):
return self._total_time return self._total_time
def toc_with_reset(self, average=True, reset_if_calls=1000): def toc_with_reset(self, average=True, reset_if_calls=1000):
""" Enable toc with reset (slightly inaccurate if reset event occurs) """ """Enable toc with reset (slightly inaccurate if reset event occurs)"""
if self._calls > reset_if_calls: if self._calls > reset_if_calls:
last_diff = time.time() - self._start_time last_diff = time.time() - self._start_time
self._start_time = time.time() self._start_time = time.time()
@ -75,7 +75,7 @@ class TimersMixin(object):
def add_timer(self, name, timer=None): def add_timer(self, name, timer=None):
if name in self._timers: if name in self._timers:
raise ValueError('timer %s already exists' % name) raise ValueError("timer %s already exists" % name)
timer = timer or Timer() timer = timer or Timer()
self._timers[name] = timer self._timers[name] = timer
return timer return timer