c87b9f49bc
- A DummyTqdm class added to utils: it replicates the interface used by trainers, but does not show the progress bar; - Added a show_progress argument to the base trainer: when show_progress == True, dummy_tqdm is used in place of tqdm.
36 lines
838 B
Python
36 lines
838 B
Python
from typing import Any
|
|
|
|
tqdm_config = {
|
|
"dynamic_ncols": True,
|
|
"ascii": True,
|
|
}
|
|
|
|
|
|
class DummyTqdm:
|
|
"""A dummy tqdm class that keeps stats but without progress bar.
|
|
|
|
It supports ``__enter__`` and ``__exit__``, update and a dummy
|
|
``set_postfix``, which is the interface that trainers use.
|
|
|
|
.. note::
|
|
|
|
Using ``disable=True`` in tqdm config results in infinite loop, thus
|
|
this class is created. See the discussion at #641 for details.
|
|
"""
|
|
|
|
def __init__(self, total: int, **kwargs: Any):
|
|
self.total = total
|
|
self.n = 0
|
|
|
|
def set_postfix(self, **kwargs: Any) -> None:
|
|
pass
|
|
|
|
def update(self, n: int = 1) -> None:
|
|
self.n += n
|
|
|
|
def __enter__(self) -> "DummyTqdm":
|
|
return self
|
|
|
|
def __exit__(self, *args: Any, **kwargs: Any) -> None:
|
|
pass
|