2021-09-03 05:05:04 +08:00
|
|
|
"""Trainer package."""
|
|
|
|
|
|
|
|
|
|
# isort:skip_file
|
|
|
|
|
|
2020-03-20 19:52:29 +08:00
|
|
|
from tianshou.trainer.utils import test_episode, gather_info
|
|
|
|
|
from tianshou.trainer.onpolicy import onpolicy_trainer
|
|
|
|
|
from tianshou.trainer.offpolicy import offpolicy_trainer
|
2021-01-20 02:13:04 -08:00
|
|
|
from tianshou.trainer.offline import offline_trainer
|
2020-03-19 17:23:46 +08:00
|
|
|
|
|
|
|
|
__all__ = [
|
2020-09-12 15:39:01 +08:00
|
|
|
"offpolicy_trainer",
|
2021-03-02 12:28:28 +08:00
|
|
|
"onpolicy_trainer",
|
2021-01-20 02:13:04 -08:00
|
|
|
"offline_trainer",
|
2021-03-02 12:28:28 +08:00
|
|
|
"test_episode",
|
|
|
|
|
"gather_info",
|
2020-03-19 17:23:46 +08:00
|
|
|
]
|
