34f714a677
Training FPS improvement (base commit is 94bfb32): test_pdqn: 1660 (without numba) -> 1930 discrete/test_ppo: 5100 -> 5170 since nstep has little impact on overall performance, the unit test result is: GAE: 4.1s -> 0.057s nstep: 0.3s -> 0.15s (little improvement) Others: - fix a bug in ttt set_eps - keep only sumtree in segment tree implementation - dirty fix for asyncVenv check_id test
18 lines
366 B
Python
18 lines
366 B
Python
from tianshou import data, env, utils, policy, trainer, exploration
|
|
|
|
# pre-compile some common-type function-call to produce the correct benchmark
|
|
# result: https://github.com/thu-ml/tianshou/pull/193#discussion_r480536371
|
|
utils.pre_compile()
|
|
|
|
|
|
__version__ = '0.2.6'
|
|
|
|
__all__ = [
|
|
'env',
|
|
'data',
|
|
'utils',
|
|
'policy',
|
|
'trainer',
|
|
'exploration',
|
|
]
|