22d7bf38c8
- rename WandBLogger -> WandbLogger - add save_data and restore_data - allow more input arguments for wandb init - integrate wandb into test/modelbase/test_psrl.py and examples/atari/atari_dqn.py - documentation update
13 lines
464 B
Python
13 lines
464 B
Python
"""Utils package."""
|
|
|
|
from tianshou.utils.config import tqdm_config
|
|
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.statistics import MovAvg, RunningMeanStd
|
|
|
|
__all__ = [
|
|
"MovAvg", "RunningMeanStd", "tqdm_config", "BaseLogger", "TensorboardLogger",
|
|
"BasicLogger", "LazyLogger", "WandbLogger"
|
|
]
|