9b61bc620c
This PR focus on refactor of logging method to solve bug of nan reward and log interval. After these two pr, hopefully fundamental change of tianshou/data is finished. We then can concentrate on building benchmarks of tianshou finally. Things changed: 1. trainer now accepts logger (BasicLogger or LazyLogger) instead of writer; 2. remove utils.SummaryWriter;
12 lines
271 B
Python
12 lines
271 B
Python
from tianshou.utils.config import tqdm_config
|
|
from tianshou.utils.moving_average import MovAvg
|
|
from tianshou.utils.log_tools import BasicLogger, LazyLogger, BaseLogger
|
|
|
|
__all__ = [
|
|
"MovAvg",
|
|
"tqdm_config",
|
|
"BaseLogger",
|
|
"BasicLogger",
|
|
"LazyLogger",
|
|
]
|