Tianshou/tianshou/trainer/__init__.py
Jialu Zhu a511cb4779
Add offline trainer and discrete BCQ algorithm (#263)
The result needs to be tuned after `done` issue fixed.

Co-authored-by: n+e <trinkle23897@gmail.com>
2021-01-20 18:13:04 +08:00

13 lines
352 B
Python

from tianshou.trainer.utils import test_episode, gather_info
from tianshou.trainer.onpolicy import onpolicy_trainer
from tianshou.trainer.offpolicy import offpolicy_trainer
from tianshou.trainer.offline import offline_trainer
__all__ = [
"gather_info",
"test_episode",
"onpolicy_trainer",
"offpolicy_trainer",
"offline_trainer",
]