Linslab commited on
Commit
74cd6e9
·
verified ·
1 Parent(s): 53f13b8

Index three completed fresh-data PI0.5 peaks with all eval50 reports and zero-success tie provenance

Browse files
README.md CHANGED
@@ -6,13 +6,15 @@ license: mit
6
 
7
  ## single_task
8
 
9
- One PI0.5 LoRA checkpoint per ManiLadder task (103 tasks), each finetuned on 20 demos.
10
  Checkpoints are selected by the highest **50-episode** success rate; ties use the earliest step.
11
 
12
  - The 60 previously published tasks use fixed evaluation seeds 1000–1049.
13
  - The 22 completed retraining runs published through 2026-10-30 use seeds 800000–800049. All reached 40,000 training steps. 15 have all eight 5k–40k evaluations; 7 have seven evaluations because one intermediate checkpoint was not retained. Their selected weights are the best among the available evaluated checkpoints.
14
  - The 21 tasks from the HF-data retraining campaign `pi05_hf21_20261025` (added 2026-10-29; raw demos freshly downloaded from Linslab/ManiLadder_Dataset) use seeds 1000–1049. All reached 40,000 training steps and have all eight 5k–40k evaluations. See [its manifest](single_task/pi05_hf21_20261025_manifest.json).
15
 
 
 
16
  See [retraining provenance and missing steps](single_task/pi05_retrain_eval50_20261022_manifest.json). These scores use different seed sets across the two groups and are not a controlled comparison.
17
 
18
  Each `single_task/<task>/` folder contains:
@@ -22,7 +24,7 @@ Each `single_task/<task>/` folder contains:
22
  - `eval_result.json` — 50-episode evaluation of this checkpoint
23
 
24
  `single_task/results.csv` lists the selected checkpoint per task; `single_task/all_checkpoints_eval.csv` has the
25
- 50-episode result of every evaluated checkpoint (814 total).
26
 
27
  Evaluation: `exec_steps=16`, `sample_steps=10`. Tasks whose checkpoints all scored 0/50 are still included;
28
  for them the step choice is arbitrary.
@@ -68,6 +70,7 @@ for them the step choice is arbitrary.
68
  | DualArmOpenLaptop-v1 | 2 | 25000 | 0.42 |
69
  | DualArmPegInHole-v1 | 2 | 5000 | 0.18 |
70
  | DualArmPlaceRopeOverRod-v1 | 2 | 40000 | 0.46 |
 
71
  | DualArmPushObject-v1 | 2 | 25000 | 0.06 |
72
  | DualArmStraightenRope-v1 | 2 | 35000 | 0.48 |
73
  | DualArmStuffCloth-v1 | 2 | 15000 | 0.40 |
@@ -85,7 +88,9 @@ for them the step choice is arbitrary.
85
  | ScoopWater-v1 | 2 | 15000 | 0.44 |
86
  | StraightenTwoRopesSequential-v1 | 2 | 15000 | 0.36 |
87
  | ClutterGrasp-v1 | 3 | 15000 | 0.16 |
 
88
  | DualArmFoldCarton-v1 | 3 | 25000 | 0.14 |
 
89
  | DualArmMakeLine-v1 | 3 | 5000 | 0.00 |
90
  | DualArmOpenFridge-v1 | 3 | 5000 | 0.00 |
91
  | DualArmOrganizeCabinet-v1 | 3 | 5000 | 0.00 |
 
6
 
7
  ## single_task
8
 
9
+ One PI0.5 LoRA checkpoint per ManiLadder task (106 tasks), each finetuned on 20 demos.
10
  Checkpoints are selected by the highest **50-episode** success rate; ties use the earliest step.
11
 
12
  - The 60 previously published tasks use fixed evaluation seeds 1000–1049.
13
  - The 22 completed retraining runs published through 2026-10-30 use seeds 800000–800049. All reached 40,000 training steps. 15 have all eight 5k–40k evaluations; 7 have seven evaluations because one intermediate checkpoint was not retained. Their selected weights are the best among the available evaluated checkpoints.
14
  - The 21 tasks from the HF-data retraining campaign `pi05_hf21_20261025` (added 2026-10-29; raw demos freshly downloaded from Linslab/ManiLadder_Dataset) use seeds 1000–1049. All reached 40,000 training steps and have all eight 5k–40k evaluations. See [its manifest](single_task/pi05_hf21_20261025_manifest.json).
15
 
16
+ - The 3 fresh-data runs in `pi05_newdata_3tasks_20261028` use pinned dataset revision `d120457e637b01679fbb40fd4de1d2394f0d55c9`, 20 demos per task, 40,000 training steps, and all eight 5k–40k eval50 checkpoints with seeds 800000–800049. **All three scored 0/50 at every checkpoint**; step 5,000 is the earliest tied checkpoint. See [their manifest](single_task/pi05_newdata_3tasks_20261028_manifest.json).
17
+
18
  See [retraining provenance and missing steps](single_task/pi05_retrain_eval50_20261022_manifest.json). These scores use different seed sets across the two groups and are not a controlled comparison.
19
 
20
  Each `single_task/<task>/` folder contains:
 
24
  - `eval_result.json` — 50-episode evaluation of this checkpoint
25
 
26
  `single_task/results.csv` lists the selected checkpoint per task; `single_task/all_checkpoints_eval.csv` has the
27
+ 50-episode result of every evaluated checkpoint (838 total).
28
 
29
  Evaluation: `exec_steps=16`, `sample_steps=10`. Tasks whose checkpoints all scored 0/50 are still included;
30
  for them the step choice is arbitrary.
 
70
  | DualArmOpenLaptop-v1 | 2 | 25000 | 0.42 |
71
  | DualArmPegInHole-v1 | 2 | 5000 | 0.18 |
72
  | DualArmPlaceRopeOverRod-v1 | 2 | 40000 | 0.46 |
73
+ | DualArmPouringTwoCups-v1 | 2 | 5000 | 0.00 |
74
  | DualArmPushObject-v1 | 2 | 25000 | 0.06 |
75
  | DualArmStraightenRope-v1 | 2 | 35000 | 0.48 |
76
  | DualArmStuffCloth-v1 | 2 | 15000 | 0.40 |
 
88
  | ScoopWater-v1 | 2 | 15000 | 0.44 |
89
  | StraightenTwoRopesSequential-v1 | 2 | 15000 | 0.36 |
90
  | ClutterGrasp-v1 | 3 | 15000 | 0.16 |
91
+ | DualArmAssembleBracket-v1 | 3 | 5000 | 0.00 |
92
  | DualArmFoldCarton-v1 | 3 | 25000 | 0.14 |
93
+ | DualArmLayerClayInDish-v1 | 3 | 5000 | 0.00 |
94
  | DualArmMakeLine-v1 | 3 | 5000 | 0.00 |
95
  | DualArmOpenFridge-v1 | 3 | 5000 | 0.00 |
96
  | DualArmOrganizeCabinet-v1 | 3 | 5000 | 0.00 |
single_task/all_checkpoints_eval.csv CHANGED
@@ -307,6 +307,14 @@ DualArmPlaceRopeOverRod-v1,level_2,25000,0.28,14,324.9540537391882
307
  DualArmPlaceRopeOverRod-v1,level_2,30000,0.38,19,296.7787297666073
308
  DualArmPlaceRopeOverRod-v1,level_2,35000,0.08,4,298.22417598265224
309
  DualArmPlaceRopeOverRod-v1,level_2,40000,0.46,23,289.2976510642795
 
 
 
 
 
 
 
 
310
  DualArmPushObject-v1,level_2,5000,0.0,0,0.7291
311
  DualArmPushObject-v1,level_2,10000,0.0,0,1.45198
312
  DualArmPushObject-v1,level_2,15000,0.0,0,3.1949400000000003
@@ -442,6 +450,14 @@ ClutterGrasp-v1,level_3,25000,0.12,6,nan
442
  ClutterGrasp-v1,level_3,30000,0.16,8,nan
443
  ClutterGrasp-v1,level_3,35000,0.14,7,nan
444
  ClutterGrasp-v1,level_3,40000,0.16,8,nan
 
 
 
 
 
 
 
 
445
  DualArmFoldCarton-v1,level_3,5000,0.0,0,6345.062264726162
446
  DualArmFoldCarton-v1,level_3,10000,0.02,1,8489.919916040897
447
  DualArmFoldCarton-v1,level_3,15000,0.04,2,9307.387320964337
@@ -450,6 +466,14 @@ DualArmFoldCarton-v1,level_3,25000,0.14,7,10516.428281509876
450
  DualArmFoldCarton-v1,level_3,30000,0.06,3,9551.720534439088
451
  DualArmFoldCarton-v1,level_3,35000,0.1,5,10555.199788849353
452
  DualArmFoldCarton-v1,level_3,40000,0.04,2,9328.819932291508
 
 
 
 
 
 
 
 
453
  DualArmMakeLine-v1,level_3,5000,0.0,0,11880.993419999999
454
  DualArmMakeLine-v1,level_3,10000,0.0,0,12168.136680000001
455
  DualArmMakeLine-v1,level_3,15000,0.0,0,12659.39848
 
307
  DualArmPlaceRopeOverRod-v1,level_2,30000,0.38,19,296.7787297666073
308
  DualArmPlaceRopeOverRod-v1,level_2,35000,0.08,4,298.22417598265224
309
  DualArmPlaceRopeOverRod-v1,level_2,40000,0.46,23,289.2976510642795
310
+ DualArmPouringTwoCups-v1,level_2,5000,0.0,0,2313.0995259738716
311
+ DualArmPouringTwoCups-v1,level_2,10000,0.0,0,2159.899790851027
312
+ DualArmPouringTwoCups-v1,level_2,15000,0.0,0,1917.7337250904739
313
+ DualArmPouringTwoCups-v1,level_2,20000,0.0,0,2533.5854001205757
314
+ DualArmPouringTwoCups-v1,level_2,25000,0.0,0,2307.9573923527146
315
+ DualArmPouringTwoCups-v1,level_2,30000,0.0,0,2610.355446477852
316
+ DualArmPouringTwoCups-v1,level_2,35000,0.0,0,2287.075251869607
317
+ DualArmPouringTwoCups-v1,level_2,40000,0.0,0,2507.2310234421384
318
  DualArmPushObject-v1,level_2,5000,0.0,0,0.7291
319
  DualArmPushObject-v1,level_2,10000,0.0,0,1.45198
320
  DualArmPushObject-v1,level_2,15000,0.0,0,3.1949400000000003
 
450
  ClutterGrasp-v1,level_3,30000,0.16,8,nan
451
  ClutterGrasp-v1,level_3,35000,0.14,7,nan
452
  ClutterGrasp-v1,level_3,40000,0.16,8,nan
453
+ DualArmAssembleBracket-v1,level_3,5000,0.0,0,404.7625858747959
454
+ DualArmAssembleBracket-v1,level_3,10000,0.0,0,610.9061906814575
455
+ DualArmAssembleBracket-v1,level_3,15000,0.0,0,863.3327129042149
456
+ DualArmAssembleBracket-v1,level_3,20000,0.0,0,582.8615937566757
457
+ DualArmAssembleBracket-v1,level_3,25000,0.0,0,1010.4638057649136
458
+ DualArmAssembleBracket-v1,level_3,30000,0.0,0,978.0950652432442
459
+ DualArmAssembleBracket-v1,level_3,35000,0.0,0,771.3933880138397
460
+ DualArmAssembleBracket-v1,level_3,40000,0.0,0,945.0748059856892
461
  DualArmFoldCarton-v1,level_3,5000,0.0,0,6345.062264726162
462
  DualArmFoldCarton-v1,level_3,10000,0.02,1,8489.919916040897
463
  DualArmFoldCarton-v1,level_3,15000,0.04,2,9307.387320964337
 
466
  DualArmFoldCarton-v1,level_3,30000,0.06,3,9551.720534439088
467
  DualArmFoldCarton-v1,level_3,35000,0.1,5,10555.199788849353
468
  DualArmFoldCarton-v1,level_3,40000,0.04,2,9328.819932291508
469
+ DualArmLayerClayInDish-v1,level_3,5000,0.0,0,421.427312732935
470
+ DualArmLayerClayInDish-v1,level_3,10000,0.0,0,419.9999928474426
471
+ DualArmLayerClayInDish-v1,level_3,15000,0.0,0,436.26779628276824
472
+ DualArmLayerClayInDish-v1,level_3,20000,0.0,0,419.9999928474426
473
+ DualArmLayerClayInDish-v1,level_3,25000,0.0,0,436.08672299027444
474
+ DualArmLayerClayInDish-v1,level_3,30000,0.0,0,426.5973575603962
475
+ DualArmLayerClayInDish-v1,level_3,35000,0.0,0,493.774410200119
476
+ DualArmLayerClayInDish-v1,level_3,40000,0.0,0,477.8500285220146
477
  DualArmMakeLine-v1,level_3,5000,0.0,0,11880.993419999999
478
  DualArmMakeLine-v1,level_3,10000,0.0,0,12168.136680000001
479
  DualArmMakeLine-v1,level_3,15000,0.0,0,12659.39848
single_task/pi05_newdata_3tasks_20261028_manifest.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "campaign": "pi05_newdata_3tasks_20261028",
3
+ "episodes_per_eval": 50,
4
+ "episode_seed_base": 800000,
5
+ "tasks": [
6
+ {
7
+ "task": "DualArmPouringTwoCups-v1",
8
+ "level": "level_2",
9
+ "suite": "deformable/dual_arm_dex",
10
+ "step": 5000,
11
+ "success_rate": 0.0,
12
+ "successes": 0,
13
+ "episodes": 50,
14
+ "episode_seed_base": 800000,
15
+ "ckpts_evaluated": 8,
16
+ "missing_eval_steps": [],
17
+ "campaign": "pi05_newdata_3tasks_20261028",
18
+ "checkpoint": "single_task/DualArmPouringTwoCups-v1/lora_step_0005000.npz",
19
+ "checkpoint_sha256": "e6f0c091d86fa5b989128fc20867d33a488c2fc3893c7adc4cba979676bec741",
20
+ "dataset_revision": "d120457e637b01679fbb40fd4de1d2394f0d55c9"
21
+ },
22
+ {
23
+ "task": "DualArmLayerClayInDish-v1",
24
+ "level": "level_3",
25
+ "suite": "deformable/dual_arm_dex",
26
+ "step": 5000,
27
+ "success_rate": 0.0,
28
+ "successes": 0,
29
+ "episodes": 50,
30
+ "episode_seed_base": 800000,
31
+ "ckpts_evaluated": 8,
32
+ "missing_eval_steps": [],
33
+ "campaign": "pi05_newdata_3tasks_20261028",
34
+ "checkpoint": "single_task/DualArmLayerClayInDish-v1/lora_step_0005000.npz",
35
+ "checkpoint_sha256": "60b9ea56c7eb9eb51ea8623541f5f4e70f6435d2ecde5fa9ce99ae58d0b611a9",
36
+ "dataset_revision": "d120457e637b01679fbb40fd4de1d2394f0d55c9"
37
+ },
38
+ {
39
+ "task": "DualArmAssembleBracket-v1",
40
+ "level": "level_3",
41
+ "suite": "rigid_body/dual_arm_gripper",
42
+ "step": 5000,
43
+ "success_rate": 0.0,
44
+ "successes": 0,
45
+ "episodes": 50,
46
+ "episode_seed_base": 800000,
47
+ "ckpts_evaluated": 8,
48
+ "missing_eval_steps": [],
49
+ "campaign": "pi05_newdata_3tasks_20261028",
50
+ "checkpoint": "single_task/DualArmAssembleBracket-v1/lora_step_0005000.npz",
51
+ "checkpoint_sha256": "162f72d171b4fa6d9511056d713dd0724de51816955f256465cb4e3a83a3761a",
52
+ "dataset_revision": "d120457e637b01679fbb40fd4de1d2394f0d55c9"
53
+ }
54
+ ],
55
+ "selection": "Highest eval50 success; earliest checkpoint breaks ties. All three tasks scored zero at all eight checkpoints.",
56
+ "evaluation_note": "TwoCups 20k\u201340k reports combine serial prefixes and validated five-episode blocks. Original prefix reward totals have three-decimal precision; successes and seeds are exact."
57
+ }
single_task/results.csv CHANGED
@@ -38,6 +38,7 @@ DualArmOpenBottle-v1,level_2,30000,0.06,3,8,single_task/DualArmOpenBottle-v1/lor
38
  DualArmOpenLaptop-v1,level_2,25000,0.42,21,8,single_task/DualArmOpenLaptop-v1/lora_step_0025000.npz
39
  DualArmPegInHole-v1,level_2,5000,0.18,9,8,single_task/DualArmPegInHole-v1/lora_step_0005000.npz
40
  DualArmPlaceRopeOverRod-v1,level_2,40000,0.46,23,8,single_task/DualArmPlaceRopeOverRod-v1/lora_step_0040000.npz
 
41
  DualArmPushObject-v1,level_2,25000,0.06,3,8,single_task/DualArmPushObject-v1/lora_step_0025000.npz
42
  DualArmStraightenRope-v1,level_2,35000,0.48,24,7,single_task/DualArmStraightenRope-v1/lora_step_0035000.npz
43
  DualArmStuffCloth-v1,level_2,15000,0.4,20,8,single_task/DualArmStuffCloth-v1/lora_step_0015000.npz
@@ -55,7 +56,9 @@ RopeBridgeDex-v1,level_2,30000,0.78,39,8,single_task/RopeBridgeDex-v1/lora_step_
55
  ScoopWater-v1,level_2,15000,0.44,22,8,single_task/ScoopWater-v1/lora_step_0015000.npz
56
  StraightenTwoRopesSequential-v1,level_2,15000,0.36,18,8,single_task/StraightenTwoRopesSequential-v1/lora_step_0015000.npz
57
  ClutterGrasp-v1,level_3,15000,0.16,8,8,single_task/ClutterGrasp-v1/lora_step_0015000.npz
 
58
  DualArmFoldCarton-v1,level_3,25000,0.14,7,8,single_task/DualArmFoldCarton-v1/lora_step_0025000.npz
 
59
  DualArmMakeLine-v1,level_3,5000,0.0,0,8,single_task/DualArmMakeLine-v1/lora_step_0005000.npz
60
  DualArmOpenFridge-v1,level_3,5000,0.0,0,8,single_task/DualArmOpenFridge-v1/lora_step_0005000.npz
61
  DualArmOrganizeCabinet-v1,level_3,5000,0.0,0,8,single_task/DualArmOrganizeCabinet-v1/lora_step_0005000.npz
 
38
  DualArmOpenLaptop-v1,level_2,25000,0.42,21,8,single_task/DualArmOpenLaptop-v1/lora_step_0025000.npz
39
  DualArmPegInHole-v1,level_2,5000,0.18,9,8,single_task/DualArmPegInHole-v1/lora_step_0005000.npz
40
  DualArmPlaceRopeOverRod-v1,level_2,40000,0.46,23,8,single_task/DualArmPlaceRopeOverRod-v1/lora_step_0040000.npz
41
+ DualArmPouringTwoCups-v1,level_2,5000,0.0,0,8,single_task/DualArmPouringTwoCups-v1/lora_step_0005000.npz
42
  DualArmPushObject-v1,level_2,25000,0.06,3,8,single_task/DualArmPushObject-v1/lora_step_0025000.npz
43
  DualArmStraightenRope-v1,level_2,35000,0.48,24,7,single_task/DualArmStraightenRope-v1/lora_step_0035000.npz
44
  DualArmStuffCloth-v1,level_2,15000,0.4,20,8,single_task/DualArmStuffCloth-v1/lora_step_0015000.npz
 
56
  ScoopWater-v1,level_2,15000,0.44,22,8,single_task/ScoopWater-v1/lora_step_0015000.npz
57
  StraightenTwoRopesSequential-v1,level_2,15000,0.36,18,8,single_task/StraightenTwoRopesSequential-v1/lora_step_0015000.npz
58
  ClutterGrasp-v1,level_3,15000,0.16,8,8,single_task/ClutterGrasp-v1/lora_step_0015000.npz
59
+ DualArmAssembleBracket-v1,level_3,5000,0.0,0,8,single_task/DualArmAssembleBracket-v1/lora_step_0005000.npz
60
  DualArmFoldCarton-v1,level_3,25000,0.14,7,8,single_task/DualArmFoldCarton-v1/lora_step_0025000.npz
61
+ DualArmLayerClayInDish-v1,level_3,5000,0.0,0,8,single_task/DualArmLayerClayInDish-v1/lora_step_0005000.npz
62
  DualArmMakeLine-v1,level_3,5000,0.0,0,8,single_task/DualArmMakeLine-v1/lora_step_0005000.npz
63
  DualArmOpenFridge-v1,level_3,5000,0.0,0,8,single_task/DualArmOpenFridge-v1/lora_step_0005000.npz
64
  DualArmOrganizeCabinet-v1,level_3,5000,0.0,0,8,single_task/DualArmOrganizeCabinet-v1/lora_step_0005000.npz