Timer#

class torch_ecg.utils.Timer(name: str | None = None, verbose: int = 0)[source]#

Bases: ReprMixin

Context manager to time the execution of a block of code.

Parameters:
  • name (str, optional) – Name of the timer, defaults to “default timer”.

  • verbose (int, default 0) – Verbosity level of the timer.

Example

>>> with Timer("task name", verbose=2) as timer:
...     do_something()
...     timer.add_time("subtask 1", level=2)
...     do_subtask_1()
...     timer.stop_timer("subtask 1")
...     timer.add_time("subtask 2", level=2)
...     do_subtask_2()
...     timer.stop_timer("subtask 2")
...     do_something_else()
add_timer(name: str, level: int = 1) None[source]#

Add a new timer for some sub-task.

Parameters:
  • name (str) – Name of the timer to be added.

  • level (int, default 1) – Verbosity level of the timer.

Return type:

None

extra_repr_keys() List[str][source]#

Extra keys for __repr__() and __str__().

stop_timer(name: str) None[source]#

Stop a timer.

Parameters:

name (str) – Name of the timer to be stopped.

Return type:

None