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

Testing kl-projection (it's working!)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-07-02 16:41:25 +02:00
parent ad7ed0071b
commit 92204c448f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test.py
View File

@ -19,7 +19,7 @@ root_path = '.'
def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=10_000_000, showRes=True, saveModel=True, n_eval_episodes=0): def main(env_name='ColumbusCandyland_Aux10-v0', timesteps=10_000_000, showRes=True, saveModel=True, n_eval_episodes=0):
env = gym.make(env_name) env = gym.make(env_name)
use_sde = True use_sde = False
ppo = PPO( ppo = PPO(
"MlpPolicy", "MlpPolicy",
env, env,
@ -101,5 +101,5 @@ def testModel(model, timesteps, showRes=False, saveModel=False, n_eval_episodes=
if __name__ == '__main__': if __name__ == '__main__':
# main('LunarLanderContinuous-v2') # main('LunarLanderContinuous-v2')
# main('ColumbusJustState-v0') # main('ColumbusJustState-v0')
# main('ColumbusStateWithBarriers-v0') main('ColumbusStateWithBarriers-v0')
main('ColumbusEasierObstacles-v0') # main('ColumbusEasierObstacles-v0')