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

[Proposal] MuJoCo Datasets #242

Open
1 task done
smorad opened this issue Sep 13, 2024 · 1 comment
Open
1 task done

[Proposal] MuJoCo Datasets #242

smorad opened this issue Sep 13, 2024 · 1 comment

Comments

@smorad
Copy link

smorad commented Sep 13, 2024

Proposal

Include the mujoco datasets from the d4rl package. Although #79 suggests this is already implemented, it doesn't seem listed on the server

 minari list remote                                                                          
                                           Minari datasets in Farama server                                           
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━┳━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━┳━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━┓
┃ Name                              ┃ Versions ┃ Total Episodes ┃ Total Steps ┃ Dataset Size ┃ Author                ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━╇━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━╇━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━┩
│ D4RL/antmaze/large-diverse-v1     │   v1, v0 │           1000 │     1000000 │ 605.2 MB     │ Alex Davey            │
│ D4RL/antmaze/large-play-v1        │   v1, v0 │           1000 │     1000000 │ 605.2 MB     │ Alex Davey            │
│ D4RL/antmaze/medium-diverse-v1    │   v1, v0 │           1000 │     1000000 │ 605.2 MB     │ Alex Davey            │
│ D4RL/antmaze/medium-play-v1       │   v1, v0 │           1000 │     1000000 │ 605.2 MB     │ Alex Davey            │
│ D4RL/antmaze/umaze-v1             │   v1, v0 │           1430 │     1000000 │ 605.0 MB     │ Alex Davey            │
│ D4RL/antmaze/umaze-diverse-v1     │   v1, v0 │           1430 │     1000000 │ 605.0 MB     │ Alex Davey            │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/door/cloned-v2               │       v2 │           4358 │     1000000 │ 532.3 MB     │ Rodrigo de Lazcano    │
│ D4RL/door/expert-v2               │       v2 │           5000 │     1000000 │ 543.3 MB     │ Rodrigo de Lazcano    │
│ D4RL/door/human-v2                │       v2 │             25 │        6729 │ 3.5 MB       │ Rodrigo de Lazcano    │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/hammer/cloned-v2             │       v2 │           3606 │     1000000 │ 564.5 MB     │ Rodrigo de Lazcano    │
│ D4RL/hammer/expert-v2             │       v2 │           5000 │     1000000 │ 584.4 MB     │ Rodrigo de Lazcano    │
│ D4RL/hammer/human-v2              │       v2 │             25 │       11310 │ 6.2 MB       │ Rodrigo de Lazcano    │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/kitchen/complete-v1          │       v1 │             19 │        4209 │ 4.3 MB       │ Rodrigo de Lazcano    │
│ D4RL/kitchen/mixed-v1             │       v1 │            621 │      156560 │ 157.5 MB     │ Rodrigo de Lazcano    │
│ D4RL/kitchen/partial-v1           │       v1 │            621 │      156560 │ 155.1 MB     │ Rodrigo de Lazcano    │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/minigrid/fourrooms-v0        │       v0 │            590 │       10010 │ 14.6 MB      │ Omar G. Younis        │
│ D4RL/minigrid/fourrooms-random-v0 │       v0 │          10174 │     1000070 │ 453.8 MB     │ Omar G. Younis        │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/pen/cloned-v2                │       v2 │           3736 │      500000 │ 313.6 MB     │ Rodrigo de Lazcano    │
│ D4RL/pen/expert-v2                │       v2 │           4958 │      499206 │ 338.3 MB     │ Rodrigo de Lazcano    │
│ D4RL/pen/human-v2                 │       v2 │             25 │        5000 │ 2.9 MB       │ Rodrigo de Lazcano    │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/pointmaze/large-v2           │   v2, v1 │           3360 │     1000000 │ 239.2 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/large-dense-v2     │   v2, v1 │           3360 │     1000000 │ 239.2 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/medium-v2          │   v2, v1 │           4752 │     1000000 │ 284.0 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/medium-dense-v2    │   v2, v1 │           4752 │     1000000 │ 284.0 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/open-v2            │   v2, v1 │           9525 │     1000000 │ 437.6 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/open-dense-v2      │   v2, v1 │           9525 │     1000000 │ 437.6 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/umaze-v2           │   v2, v1 │          13210 │     1000000 │ 556.2 MB     │ Rodrigo Perez-Vicente │
│ D4RL/pointmaze/umaze-dense-v2     │   v2, v1 │          13210 │     1000000 │ 556.2 MB     │ Rodrigo Perez-Vicente │
├───────────────────────────────────┼──────────┼────────────────┼─────────────┼──────────────┼───────────────────────┤
│ D4RL/relocate/cloned-v2           │       v2 │           3758 │     1000000 │ 527.7 MB     │ Rodrigo de Lazcano    │
│ D4RL/relocate/expert-v2           │       v2 │           5000 │     1000000 │ 552.2 MB     │ Rodrigo de Lazcano    │
│ D4RL/relocate/human-v2            │       v2 │             25 │        9942 │ 5.0 MB       │ Rodrigo de Lazcano    │
└───────────────────────────────────┴──────────┴────────────────┴─────────────┴──────────────┴───────────────────────┘

Motivation

These are probably the most-used environments, it would be nice if we could download such datasets through minari.

Pitch

minari list remote should list reacher, hopper, half-cheetah etc.

Checklist

  • I have checked that there is no similar issue in the repo (required)
@younik
Copy link
Member

younik commented Sep 14, 2024

We are planning to add it as soon as Gymnasium 1.0 is out; which I believe will happen soon.

We have the expert policies, here you can find the script for generating them:
https://github.com/Farama-Foundation/minari-dataset-generation-scripts/pull/8/files

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

No branches or pull requests

2 participants