iFaz commited on
Commit
d344823
·
verified ·
1 Parent(s): 07ca882

Add auto-generated model card from override dicts

Browse files
Files changed (1) hide show
  1. README.md +60 -35
README.md CHANGED
@@ -1,62 +1,87 @@
1
  ---
2
- datasets: lerobot/aloha_sim_transfer_cube_human
 
 
3
  library_name: lerobot
4
  license: apache-2.0
5
- model_name: diffusion
6
- pipeline_tag: robotics
7
  tags:
8
- - lerobot
9
  - robotics
 
10
  - diffusion
 
 
11
  ---
12
 
13
- # Model Card for diffusion
14
 
15
- <!-- Provide a quick summary of what the model is/does. -->
 
 
16
 
 
 
 
 
 
 
 
17
 
18
- [Diffusion Policy](https://huggingface.co/papers/2303.04137) treats visuomotor control as a generative diffusion process, producing smooth, multi-step action trajectories that excel at contact-rich manipulation.
19
 
 
20
 
21
- This policy has been trained and pushed to the Hub using [LeRobot](https://github.com/huggingface/lerobot).
22
- See the full documentation at [LeRobot Docs](https://huggingface.co/docs/lerobot/index).
 
 
 
 
 
 
 
 
 
23
 
24
  ---
25
 
26
- ## How to Get Started with the Model
27
 
28
- For a complete walkthrough, see the [training guide](https://huggingface.co/docs/lerobot/il_robots#train-a-policy).
29
- Below is the short version on how to train and run inference/eval:
30
 
31
- ### Train from scratch
32
 
33
- ```bash
34
- lerobot-train \
35
- --dataset.repo_id=${HF_USER}/<dataset> \
36
- --policy.type=act \
37
- --output_dir=outputs/train/<desired_policy_repo_id> \
38
- --job_name=lerobot_training \
39
- --policy.device=cuda \
40
- --policy.repo_id=${HF_USER}/<desired_policy_repo_id>
41
- --wandb.enable=true
42
- ```
43
 
44
- _Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
 
 
 
 
 
 
 
45
 
46
- ### Evaluate the policy/run inference
47
 
48
- ```bash
49
- lerobot-record \
50
- --robot.type=so100_follower \
51
- --dataset.repo_id=<hf_user>/eval_<dataset> \
52
- --policy.path=<hf_user>/<desired_policy_repo_id> \
53
- --episodes=10
54
- ```
55
 
56
- Prefix the dataset repo with **eval\_** and supply `--policy.path` pointing to a local or hub checkpoint.
 
 
 
 
 
 
57
 
58
  ---
59
 
60
- ## Model Details
61
 
62
- - **License:** apache-2.0
 
 
 
 
 
 
 
 
1
  ---
2
+ datasets:
3
+ - lerobot/aloha_sim_transfer_cube_human
4
+ language: en
5
  library_name: lerobot
6
  license: apache-2.0
 
 
7
  tags:
 
8
  - robotics
9
+ - imitation-learning
10
  - diffusion
11
+ - mujoco
12
+ - pytorch_model_hub_mixin
13
  ---
14
 
15
+ # DIFFUSION Policy diffusion_aloha_transfer_cube_seed42
16
 
17
+ Trained with [LeRobot](https://github.com/huggingface/lerobot).
18
+ Date: `2026-05-14 12:51`
19
+ Policy type: `diffusion` | Device: `cuda`
20
 
21
+ ---
22
+
23
+ ## 📦 Dataset
24
+
25
+ | Parameter | Value |
26
+ |---|---|
27
+ | `dataset.repo_id` | `lerobot/aloha_sim_transfer_cube_human` |
28
 
29
+ ---
30
 
31
+ ## 🏋️ Training Config
32
 
33
+ | Parameter | Value |
34
+ |---|---|
35
+ | `steps` | `1000` |
36
+ | `batch_size` | `8` |
37
+ | `eval_freq` | `500` |
38
+ | `save_freq` | `500` |
39
+ | `num_workers` | `4` |
40
+ | `seed` | `42` |
41
+ | `eval.n_episodes` | `1` |
42
+ | `eval.batch_size` | `1` |
43
+ | `eval.use_async_envs` | `False` |
44
 
45
  ---
46
 
47
+ ## 📐 Policy Architecture
48
 
49
+ _All defaults no overrides applied._
 
50
 
51
+ ---
52
 
53
+ ## 🎯 Eval Config
 
 
 
 
 
 
 
 
 
54
 
55
+ | Parameter | Value |
56
+ |---|---|
57
+ | `env.type` | `aloha` |
58
+ | `env.task` | `AlohaTransferCube-v0` |
59
+ | `eval.n_episodes` | `5` |
60
+ | `eval.batch_size` | `1` |
61
+ | `eval.use_async_envs` | `False` |
62
+ | `policy.path` | `/kaggle/working/outputs/train/aloha_transfer_cube_seed42/checkpoints/last/pretrained_model` |
63
 
64
+ ---
65
 
66
+ ## 📊 Eval Results
 
 
 
 
 
 
67
 
68
+ | Metric | Value |
69
+ |---|---|
70
+ | Episodes | `5` |
71
+ | Success rate | `0.0%` |
72
+ | Avg sum reward | `0.00` |
73
+ | Avg max reward | `0.00` |
74
+ | Eval time (s) | `739.9` |
75
 
76
  ---
77
 
78
+ ## Citation
79
 
80
+ ```bibtex
81
+ @misc{cadene2024lerobot,
82
+ author = {Cadene, Remi and Alibert, Simon and others},
83
+ title = {LeRobot},
84
+ year = {2024},
85
+ url = {https://github.com/huggingface/lerobot}
86
+ }
87
+ ```