Tianshou/tianshou/__init__.py
n+e 710966eda7
change API of train_fn and test_fn (#229)
train_fn(epoch) -> train_fn(epoch, num_env_step)
test_fn(epoch) -> test_fn(epoch, num_env_step)
2020-09-26 16:35:37 +08:00

14 lines
191 B
Python

from tianshou import data, env, utils, policy, trainer, exploration
__version__ = "0.3.0"
__all__ = [
"env",
"data",
"utils",
"policy",
"trainer",
"exploration",
]