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

Add MuJoCo environments support. #76

Merged
merged 1 commit into from
Nov 1, 2021
Merged

Add MuJoCo environments support. #76

merged 1 commit into from
Nov 1, 2021

Conversation

vwxyzjn
Copy link
Owner

@vwxyzjn vwxyzjn commented Oct 30, 2021

This PR adds support for MuJoCo environments such as Hopper-v2. Only works for Linux.

poetry install -E mujoco
poetry install -E pybullet
poetry run python cleanrl/ppo_continuous_action.py --gym-id Hopper-v2
poetry run python cleanrl/ppo_continuous_action.py --gym-id HalfCheetahBulletEnv-v0

@vwxyzjn vwxyzjn requested a review from dosssman October 30, 2021 01:18
Copy link
Collaborator

@dosssman dosssman left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Mujoco integration working, tested with ppo_continuous_action.py and sac_continuous_actions.py.

I notice among the file changed that there are quite a few brax related files.
Are those really part and / or relevant to this PR ?

@vwxyzjn
Copy link
Owner Author

vwxyzjn commented Nov 1, 2021

Yeah the brax stuff comes from another branch... Let me remove them.

@vwxyzjn vwxyzjn merged commit 96281d8 into master Nov 1, 2021
@vwxyzjn vwxyzjn deleted the mujuco branch November 1, 2021 13:22
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.

None yet

2 participants