dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

LunarLanderContinuous-v2 is our new default test-env

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-25 21:47:21 +02:00
parent cf5a2e82fc
commit 60c954c8c1
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test.py
View File

@ -17,7 +17,7 @@ import columbus
root_path = '.' root_path = '.'
def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=5000, showRes=False, saveModel=True, n_eval_episodes=8): def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False, saveModel=True, n_eval_episodes=16):
env = gym.make(env_name) env = gym.make(env_name)
test_sde = False test_sde = False
ppo = PPO( ppo = PPO(
@ -87,12 +87,12 @@ def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=5000, showRes=False, s
# learning_rate=0.001 # 0.015 # learning_rate=0.001 # 0.015
# ) # )
print('PPO:')
testModel(ppo, timesteps, showRes,
saveModel, n_eval_episodes)
print('TRL_PG:') print('TRL_PG:')
testModel(trl_pg, timesteps, showRes, testModel(trl_pg, timesteps, showRes,
saveModel, n_eval_episodes) saveModel, n_eval_episodes)
print('PPO:')
testModel(ppo, timesteps, showRes,
saveModel, n_eval_episodes)
def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=16): def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=16):
@ -131,4 +131,4 @@ def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=
if __name__ == '__main__': if __name__ == '__main__':
main('CartPole-v1') main('LunarLanderContinuous-v2')