Tianshou/tianshou/trainer/__init__.py
2020-03-20 19:52:29 +08:00

11 lines
276 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
__all__ = [
'gather_info',
'test_episode',
'onpolicy_trainer',
'offpolicy_trainer',
]