Skip to content
Snippets Groups Projects
Commit fb37266a authored by oscarchaufour's avatar oscarchaufour
Browse files

Update a2c_sb3_panda_reach.py

parent 6653d969
No related branches found
No related tags found
No related merge requests found
...@@ -17,17 +17,17 @@ total_timesteps = 100000 ...@@ -17,17 +17,17 @@ total_timesteps = 100000
config = { config = {
"policy_type": "MlpPolicy", "policy_type": "MlpPolicy",
"total_timesteps": total_timesteps, "total_timesteps": total_timesteps,
"env_name": "CartPole-v1", "env_name": "PandaReachJointsDense-v3",
} }
wandb.login() wandb.login()
run = wandb.init( run = wandb.init(
project="a2c-PandaReachJointsDense-v2", project="a2c-PandaReachJointsDense-v3",
config=config, config=config,
sync_tensorboard=True, # auto-upload sb3's tensorboard metrics sync_tensorboard=True, # auto-upload sb3's tensorboard metrics
monitor_gym=True, # auto-upload the videos of agents playing the game monitor_gym=True, # auto-upload the videos of agents playing the game
save_code=True, # optional save_code=True, # optional
) )
env_id = "PandaReachJointsDense-v2" env_id = "PandaReachJointsDense-v3"
# Register the environment # Register the environment
register(id=env_id, entry_point='gym.envs.classic_control:CartPoleEnv', max_episode_steps=500) register(id=env_id, entry_point='gym.envs.classic_control:CartPoleEnv', max_episode_steps=500)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment