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

Dataset generation for MuJoCo environments #17

Merged
merged 27 commits into from
Jan 28, 2025

Conversation

Kallinteris-Andreas
Copy link
Collaborator

@Kallinteris-Andreas Kallinteris-Andreas commented Oct 21, 2024

Implements: Farama-Foundation/Minari#243

Does not include generation scripts for HumanoidStandup Because I have not generated good enough agents yet

based on #8

  • evaluate_policy.py is to evaluate rodrigodelazcano's policies
  • create_dataset.py should be self-explanatory
  • train.py if you starting read the file from __main__ it is easier to read,

wandb: https://wandb.ai/kallinteris/Minari/overview
evaluation repo: https://github.com/Kallinteris-Andreas/gymnasium-mujoco-minari-v0-datasets
also closes: #8

@Kallinteris-Andreas Kallinteris-Andreas changed the title Dataset generation for MuJoCo environments Dataset generation for MuJoCo environments Oct 21, 2024
Copy link
Member

@younik younik left a comment

Choose a reason for hiding this comment

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

Many thanks @Kallinteris-Andreas

I left some comments

scripts/mujoco/create_dataset.py Outdated Show resolved Hide resolved
scripts/mujoco/create_dataset.py Outdated Show resolved Hide resolved
scripts/mujoco/create_dataset.py Outdated Show resolved Hide resolved
scripts/mujoco/create_dataset.py Outdated Show resolved Hide resolved
scripts/mujoco/create_dataset_humanoid_standup.py Outdated Show resolved Hide resolved
scripts/mujoco/create_dataset_humanoid_standup.py Outdated Show resolved Hide resolved
scripts/mujoco/evaluate_dataset.py Outdated Show resolved Hide resolved
scripts/mujoco/evaluate_policy.py Outdated Show resolved Hide resolved
scripts/mujoco/requirements.txt Outdated Show resolved Hide resolved
scripts/train.py Show resolved Hide resolved
@Kallinteris-Andreas
Copy link
Collaborator Author

@younik
I uploaded the final version of the scripts (hopefully final), It is a lot cleaner, and includes datasets descriptions

the models are ready and uploaded to HF, the datasets are ready for uploading

Only Swimmer and Humanoid are missing, which should be coming soon

Copy link
Member

@younik younik left a comment

Choose a reason for hiding this comment

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

Thank you!
It looks good, if you have the descriptions, please update them as well.
Feel free to merge it after it

author="Kallinteris Andreas",
author_email="[email protected]",
requirements=["mujoco==3.2.3", "gymnasium>=1.0.0"],
description=open(f"./descriptions/{env_id}-{proficiency}.md", 'r').read()
Copy link
Member

Choose a reason for hiding this comment

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

do you also have these descriptions?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

oops i forgot to git add them

@Kallinteris-Andreas Kallinteris-Andreas merged commit c489a90 into Farama-Foundation:main Jan 28, 2025
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