8a5e2190f7
- rename BasicLogger to TensorboardLogger - refactor logger code - add WandbLogger Co-authored-by: Jiayi Weng <trinkle23897@gmail.com>
18 lines
467 B
Python
18 lines
467 B
Python
from tianshou.utils.config import tqdm_config
|
|
from tianshou.utils.statistics import MovAvg, RunningMeanStd
|
|
from tianshou.utils.logger.base import BaseLogger, LazyLogger
|
|
from tianshou.utils.logger.tensorboard import TensorboardLogger, BasicLogger
|
|
from tianshou.utils.logger.wandb import WandBLogger
|
|
|
|
|
|
__all__ = [
|
|
"MovAvg",
|
|
"RunningMeanStd",
|
|
"tqdm_config",
|
|
"BaseLogger",
|
|
"TensorboardLogger",
|
|
"BasicLogger",
|
|
"LazyLogger",
|
|
"WandBLogger"
|
|
]
|