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

Update a2c_sb3_panda_reach.py

parent 6653d969
Branches
No related tags found
No related merge requests found
......@@ -17,17 +17,17 @@ total_timesteps = 100000
config = {
"policy_type": "MlpPolicy",
"total_timesteps": total_timesteps,
"env_name": "CartPole-v1",
"env_name": "PandaReachJointsDense-v3",
}
wandb.login()
run = wandb.init(
project="a2c-PandaReachJointsDense-v2",
project="a2c-PandaReachJointsDense-v3",
config=config,
sync_tensorboard=True, # auto-upload sb3's tensorboard metrics
monitor_gym=True, # auto-upload the videos of agents playing the game
save_code=True, # optional
)
env_id = "PandaReachJointsDense-v2"
env_id = "PandaReachJointsDense-v3"
# Register the environment
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