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

StillTesting

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-26 16:38:46 +02:00
parent 80741776d2
commit 024a9a0265
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test.py
View File

@ -17,7 +17,7 @@ import columbus
root_path = '.' root_path = '.'
def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False, saveModel=True, n_eval_episodes=16): def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=500000, showRes=True, saveModel=True, n_eval_episodes=0):
env = gym.make(env_name) env = gym.make(env_name)
test_sde = False test_sde = False
ppo = PPO( ppo = PPO(
@ -29,7 +29,7 @@ def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False,
gamma=0.99, gamma=0.99,
gae_lambda=0.95, gae_lambda=0.95,
normalize_advantage=True, normalize_advantage=True,
ent_coef=0.15, # 0.1 ent_coef=0.1, # 0.1
vf_coef=0.5, vf_coef=0.5,
use_sde=False, # False use_sde=False, # False
) )
@ -42,7 +42,7 @@ def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False,
gamma=0.99, gamma=0.99,
gae_lambda=0.95, gae_lambda=0.95,
normalize_advantage=True, normalize_advantage=True,
ent_coef=0.15, # 0.1 ent_coef=0.1, # 0.1
vf_coef=0.5, vf_coef=0.5,
use_sde=False, # False use_sde=False, # False
) )
@ -90,9 +90,9 @@ def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=50000, showRes=False,
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:') #print('PPO:')
testModel(ppo, timesteps, showRes, #testModel(ppo, timesteps, showRes,
saveModel, n_eval_episodes) # 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):