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

Trying to get ColumbusEnv to work...

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-19 15:59:55 +02:00
parent f950d1fe64
commit c45e3627ea
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test.py
View File

@ -10,14 +10,15 @@ from sb3_trl.trl_pg import TRL_PG
from subtrees.columbus import env from subtrees.columbus import env
register( register(
id='Columbus_Test3.1-v0', id='ColumbusTest3.1-v0',
entry_point=env.ColumbusEnv, entry_point=env.ColumbusTest3_1,
max_episode_steps=1000, max_episode_steps=1000,
) )
def main(): def main():
#env = gym.make("LunarLander-v2") #env = gym.make("LunarLander-v2")
env = gym.make("Columbus_test3.1-v0") env = gym.make("ColumbusTest3.1-v0")
ppo = PPO( ppo = PPO(
"MlpPolicy", "MlpPolicy",
env, env,