Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added MS Ant Walk and Run Tasks #533

Merged
merged 5 commits into from
Aug 29, 2024
Merged

Conversation

Xander-Hinrichsen
Copy link
Collaborator

MS Ant walking at 0.5 m/s and running at 4 m/s tasks

antwalk.mp4

image

command to run (or use example in ppo examples.sh):

for i in {1..3}; do python ppo.py --exp_name="antwalkseed${i}" --env_id="MS-AntWalk-v1" --num_envs=2048 --update_epochs=8 --num_minibatches=32 --total_timesteps=9_000_000 --eval_freq=10 --num_eval_steps=1000 --num_steps=200 --gamma=0.97 --ent_coef=1e-3 --seed=${i}; done

antrun.mp4

image

command to run (or use example in ppo examples.sh)::

for i in {1..3}; do python ppo.py --exp_name="antrunseed${i}" --env_id="MS-AntRun-v1" --num_envs=2048 --update_epochs=8 --num_minibatches=32 --total_timesteps=30_000_000 --eval_freq=10 --num_eval_steps=1000 --num_steps=200 --gamma=0.97 --ent_coef=1e-3 --seed=${i}; done

@StoneT2000
Copy link
Member

@Xander-Hinrichsen any rgb results? seems quite doable given it solves in 1M steps

@Xander-Hinrichsen
Copy link
Collaborator Author

Xander-Hinrichsen commented Aug 29, 2024

ant walk rgb

image

ant_walk_rgb.mp4

command (already added to ppo examples.sh):
python ppo_rgb.py --env_id="MS-AntWalk-v1"
--num_envs=256 --update_epochs=8 --num_minibatches=32
--total_timesteps=5_000_000 --eval_freq=15 --num_eval_steps=1000
--num_steps=200 --gamma=0.97 --no-include-state --render_mode="rgb_array"
--ent_coef=1e-3

ant run rgb

image

ant_run_rgb.mp4

command (already added to ppo examples.sh):
python ppo_rgb.py --env_id="MS-AntRun-v1"
--num_envs=256 --update_epochs=8 --num_minibatches=32
--total_timesteps=15_000_000 --eval_freq=15 --num_eval_steps=1000
--num_steps=200 --gamma=0.97 --no-include-state --render_mode="rgb_array"
--ent_coef=1e-3

edit: all evals are 33.33 seconds long, trimmed here to fit file size max on a comment

@StoneT2000 StoneT2000 merged commit 334fadf into haosulab:main Aug 29, 2024
@Xander-Hinrichsen Xander-Hinrichsen deleted the MS-Ant branch August 29, 2024 17:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants