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.
24 lines
678 B
Python
24 lines
678 B
Python
"""Utils package."""
|
|
|
|
from tianshou.utils.logger.base import BaseLogger, LazyLogger
|
|
from tianshou.utils.logger.tensorboard import BasicLogger, TensorboardLogger
|
|
from tianshou.utils.logger.wandb import WandbLogger
|
|
from tianshou.utils.lr_scheduler import MultipleLRSchedulers
|
|
from tianshou.utils.progress_bar import DummyTqdm, tqdm_config
|
|
from tianshou.utils.statistics import MovAvg, RunningMeanStd
|
|
from tianshou.utils.warning import deprecation
|
|
|
|
__all__ = [
|
|
"MovAvg",
|
|
"RunningMeanStd",
|
|
"tqdm_config",
|
|
"DummyTqdm",
|
|
"BaseLogger",
|
|
"TensorboardLogger",
|
|
"BasicLogger",
|
|
"LazyLogger",
|
|
"WandbLogger",
|
|
"deprecation",
|
|
"MultipleLRSchedulers",
|
|
]
|