alr.training.progress_bar.ignite_progress_bar

Classses

ProgressBar

class alr.training.progress_bar.ignite_progress_bar.ProgressBar(desc: Callable = None, log_interval: Optional[int] = 10)[source]

Bases: object

Creates a smart progress bar (tqdm if in notebooks, text if in terminal). Use alr.training.progress_bar.use_tqdm = True/False to force TQDM (or force disable it).

Parameters:
  • desc (Callable, optional) – takes an engine as input and returns a string
  • log_interval (int, optional) – log every log_interval iterations
attach(engine: ignite.engine.engine.Engine)[source]
log_message(msg: str)[source]
on_complete(_)[source]
on_iteration_complete(engine)[source]
on_start(engine)[source]

Functions