Whalswp commited on
Commit
53c750a
·
verified ·
1 Parent(s): 2228f17

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. Abs_6D + DeltaEE/Baseline/checkpoints/030000/action_space_manifest.json +205 -0
  2. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/action_space_manifest.json +205 -0
  3. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/config.json +107 -0
  4. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
  5. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_preprocessor.json +78 -0
  6. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/prompt_manifest.json +40 -0
  7. Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/train_config.json +258 -0
  8. Abs_6D + DeltaEE/Baseline/checkpoints/030000/prompt_manifest.json +40 -0
  9. Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/optimizer_param_groups.json +577 -0
  10. Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/scheduler_state.json +18 -0
  11. Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/training_step.json +5 -0
  12. Abs_6D + DeltaEE/Baseline/wandb/debug-internal.log.rerun_20260719_231636 +0 -0
  13. Abs_6D + DeltaEE/Baseline/wandb/debug.log.rerun_20260719_231636 +19 -0
  14. Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/files/wandb-metadata.json +91 -0
  15. Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug-internal.log +0 -0
  16. Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug.log +19 -0
  17. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/phase_schedule.json +55 -0
  18. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/action_space_manifest.json +205 -0
  19. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/config.json +117 -0
  20. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/phase_schedule.json +55 -0
  21. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
  22. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_preprocessor.json +82 -0
  23. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/train_config.json +268 -0
  24. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/source_config.yaml +65 -0
  25. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/optimizer_param_groups.json +585 -0
  26. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/scheduler_state.json +18 -0
  27. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_175602-r3cz38bm/logs/debug-internal.log +18 -0
  28. Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_181953-a9vyia29/logs/debug-internal.log +15 -0
  29. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/requirements.txt +117 -0
  30. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/wandb-metadata.json +91 -0
  31. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/config.yaml +333 -0
  32. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/requirements.txt +117 -0
  33. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-metadata.json +52 -0
  34. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-summary.json +1 -0
  35. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-core.log +45 -0
  36. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-internal.log +18 -0
  37. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/config.yaml +332 -0
  38. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/wandb-metadata.json +52 -0
  39. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/logs/debug-core.log +48 -0
  40. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/config.yaml +332 -0
  41. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/output.log +33 -0
  42. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/wandb-summary.json +1 -0
  43. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/logs/debug.log +46 -0
  44. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/output.log +21 -0
  45. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/requirements.txt +117 -0
  46. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/wandb-metadata.json +52 -0
  47. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug-core.log +16 -0
  48. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug.log +20 -0
  49. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_182104-4bcz2tkp/files/output.log +72 -0
  50. Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_182104-4bcz2tkp/files/requirements.txt +117 -0
Abs_6D + DeltaEE/Baseline/checkpoints/030000/action_space_manifest.json ADDED
@@ -0,0 +1,205 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 2,
3
+ "action_space": "ee_abs_rot6d+ee_delta",
4
+ "action_views": [
5
+ "ee_abs_rot6d",
6
+ "ee_delta"
7
+ ],
8
+ "source_feature": null,
9
+ "canonical_feature": "action",
10
+ "dtype": "float32",
11
+ "shape": [
12
+ 16
13
+ ],
14
+ "names": [
15
+ "ee_abs_rot6d.x",
16
+ "ee_abs_rot6d.y",
17
+ "ee_abs_rot6d.z",
18
+ "ee_abs_rot6d.r_col0_x",
19
+ "ee_abs_rot6d.r_col0_y",
20
+ "ee_abs_rot6d.r_col0_z",
21
+ "ee_abs_rot6d.r_col1_x",
22
+ "ee_abs_rot6d.r_col1_y",
23
+ "ee_abs_rot6d.r_col1_z",
24
+ "ee_abs_rot6d.gripper",
25
+ "ee_delta.delta_x",
26
+ "ee_delta.delta_y",
27
+ "ee_delta.delta_z",
28
+ "ee_delta.delta_axis_angle_x",
29
+ "ee_delta.delta_axis_angle_y",
30
+ "ee_delta.delta_axis_angle_z"
31
+ ],
32
+ "dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
33
+ "dataset_revision": "v3.0",
34
+ "dataset_total_frames": 218367,
35
+ "dataset_total_episodes": 4930,
36
+ "dataset_fps": 10,
37
+ "stats_sha256": "75b95092e5fd9a1b16aa6dc7a478883927d7c7cce76c26b56b290b675ae2311a",
38
+ "stats": {
39
+ "min": [
40
+ -0.21784618496894836,
41
+ -0.4474811851978302,
42
+ 0.1702737659215927,
43
+ -0.9999986290931702,
44
+ -0.9999991655349731,
45
+ -0.9999999403953552,
46
+ -0.9999990463256836,
47
+ -0.9999998211860657,
48
+ -0.9999988079071045,
49
+ 0.0,
50
+ -0.09827068448066711,
51
+ -0.13628354668617249,
52
+ -0.12191051989793777,
53
+ -2.431101083755493,
54
+ -0.3816019296646118,
55
+ -0.5094270706176758
56
+ ],
57
+ "max": [
58
+ 0.6283698678016663,
59
+ 0.4420371949672699,
60
+ 1.051027774810791,
61
+ 0.9999999403953552,
62
+ 0.9999995231628418,
63
+ 1.0,
64
+ 0.9999995231628418,
65
+ 1.0,
66
+ 0.9999996423721313,
67
+ 0.03999999910593033,
68
+ 0.11647262424230576,
69
+ 0.13701310753822327,
70
+ 0.13330328464508057,
71
+ 1.7823294401168823,
72
+ 0.5954466462135315,
73
+ 0.6865788102149963
74
+ ],
75
+ "mean": [
76
+ 0.4040228037735655,
77
+ -0.05247618696989652,
78
+ 0.6993459771218912,
79
+ 0.07011111991957117,
80
+ -0.03520558904460197,
81
+ 0.14470242846953205,
82
+ 0.05491166606739672,
83
+ -0.19073058893378805,
84
+ -0.06988117661871228,
85
+ 0.021167300266524632,
86
+ 0.0018730097199051647,
87
+ -0.0015708771114029225,
88
+ -0.001553903484685181,
89
+ 0.014420973441752116,
90
+ 0.00961362082315661,
91
+ -0.0017634798849202702
92
+ ],
93
+ "std": [
94
+ 0.11861927913357073,
95
+ 0.11318943557281885,
96
+ 0.17849160646987516,
97
+ 0.4812528791886511,
98
+ 0.7487196258736938,
99
+ 0.4251126804030219,
100
+ 0.47797229807769137,
101
+ 0.6267695346407992,
102
+ 0.5782954306815554,
103
+ 0.01996590431050817,
104
+ 0.015762732601462,
105
+ 0.010587306860662312,
106
+ 0.01180823404609393,
107
+ 0.10086023299718701,
108
+ 0.04747329757360547,
109
+ 0.14801097710214692
110
+ ],
111
+ "count": [
112
+ 218367
113
+ ],
114
+ "q01": [
115
+ 0.23988881827972064,
116
+ -0.10387461883034811,
117
+ 0.6297243923476871,
118
+ -0.43622841955353464,
119
+ -0.6885486527617446,
120
+ -0.14103555388039019,
121
+ -0.4327210063987595,
122
+ -0.9304790915716145,
123
+ -0.3465462258490417,
124
+ -1.000000013351432e-10,
125
+ -0.02190519006673929,
126
+ -0.02092386152118027,
127
+ -0.026422736164554015,
128
+ -0.11497312558112925,
129
+ -0.025678747253255536,
130
+ -0.17124583807766966
131
+ ],
132
+ "q10": [
133
+ 0.3042306436755216,
134
+ -0.09609991620622624,
135
+ 0.6394063072781145,
136
+ -0.3351306352998669,
137
+ -0.5751963015002924,
138
+ -0.08490621283307866,
139
+ -0.33849701676224275,
140
+ -0.8340557712529717,
141
+ -0.29760290790884514,
142
+ -1.000000013351432e-10,
143
+ -0.013015102506385306,
144
+ -0.011160732577261764,
145
+ -0.012954131824952583,
146
+ -0.056165101267083004,
147
+ -0.01548487039073523,
148
+ -0.15972996251089885
149
+ ],
150
+ "q50": [
151
+ 0.4160225263528291,
152
+ -0.06267459152989589,
153
+ 0.6811777491060801,
154
+ 0.0543536902230377,
155
+ -0.14012251058819217,
156
+ 0.04772394280004727,
157
+ 0.03838119138716219,
158
+ -0.14865169094195635,
159
+ -0.2073069401172425,
160
+ 0.020772594934870346,
161
+ 0.0005058764599893245,
162
+ 4.561956601090609e-06,
163
+ 4.945756498042436e-05,
164
+ 0.0001528964582560528,
165
+ 0.00017815767540355586,
166
+ -0.0002718872707462707
167
+ ],
168
+ "q90": [
169
+ 0.46641008755679864,
170
+ 0.020233143101890064,
171
+ 0.8095135405259374,
172
+ 0.48172401592474906,
173
+ 0.7398737043070319,
174
+ 0.5970841215636613,
175
+ 0.4674752875999555,
176
+ 0.36281754777672676,
177
+ 0.5033361883834496,
178
+ 0.039998362213352394,
179
+ 0.016358020650084733,
180
+ 0.003987531854121596,
181
+ 0.006191431456332113,
182
+ 0.11275584021509974,
183
+ 0.06100667096893492,
184
+ 0.15292656959583054
185
+ ],
186
+ "q99": [
187
+ 0.47320723171849244,
188
+ 0.04114262476135705,
189
+ 0.8467967259270389,
190
+ 0.5867781409980988,
191
+ 0.8529584859257228,
192
+ 0.7953936054341789,
193
+ 0.5768269858732034,
194
+ 0.49353349323389695,
195
+ 0.7168914440776504,
196
+ 0.03999983541667277,
197
+ 0.03626038003704021,
198
+ 0.008814053033445103,
199
+ 0.012299866693624668,
200
+ 0.17731576759283033,
201
+ 0.12117986440120965,
202
+ 0.16474415715691368
203
+ ]
204
+ }
205
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/action_space_manifest.json ADDED
@@ -0,0 +1,205 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 2,
3
+ "action_space": "ee_abs_rot6d+ee_delta",
4
+ "action_views": [
5
+ "ee_abs_rot6d",
6
+ "ee_delta"
7
+ ],
8
+ "source_feature": null,
9
+ "canonical_feature": "action",
10
+ "dtype": "float32",
11
+ "shape": [
12
+ 16
13
+ ],
14
+ "names": [
15
+ "ee_abs_rot6d.x",
16
+ "ee_abs_rot6d.y",
17
+ "ee_abs_rot6d.z",
18
+ "ee_abs_rot6d.r_col0_x",
19
+ "ee_abs_rot6d.r_col0_y",
20
+ "ee_abs_rot6d.r_col0_z",
21
+ "ee_abs_rot6d.r_col1_x",
22
+ "ee_abs_rot6d.r_col1_y",
23
+ "ee_abs_rot6d.r_col1_z",
24
+ "ee_abs_rot6d.gripper",
25
+ "ee_delta.delta_x",
26
+ "ee_delta.delta_y",
27
+ "ee_delta.delta_z",
28
+ "ee_delta.delta_axis_angle_x",
29
+ "ee_delta.delta_axis_angle_y",
30
+ "ee_delta.delta_axis_angle_z"
31
+ ],
32
+ "dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
33
+ "dataset_revision": "v3.0",
34
+ "dataset_total_frames": 218367,
35
+ "dataset_total_episodes": 4930,
36
+ "dataset_fps": 10,
37
+ "stats_sha256": "75b95092e5fd9a1b16aa6dc7a478883927d7c7cce76c26b56b290b675ae2311a",
38
+ "stats": {
39
+ "min": [
40
+ -0.21784618496894836,
41
+ -0.4474811851978302,
42
+ 0.1702737659215927,
43
+ -0.9999986290931702,
44
+ -0.9999991655349731,
45
+ -0.9999999403953552,
46
+ -0.9999990463256836,
47
+ -0.9999998211860657,
48
+ -0.9999988079071045,
49
+ 0.0,
50
+ -0.09827068448066711,
51
+ -0.13628354668617249,
52
+ -0.12191051989793777,
53
+ -2.431101083755493,
54
+ -0.3816019296646118,
55
+ -0.5094270706176758
56
+ ],
57
+ "max": [
58
+ 0.6283698678016663,
59
+ 0.4420371949672699,
60
+ 1.051027774810791,
61
+ 0.9999999403953552,
62
+ 0.9999995231628418,
63
+ 1.0,
64
+ 0.9999995231628418,
65
+ 1.0,
66
+ 0.9999996423721313,
67
+ 0.03999999910593033,
68
+ 0.11647262424230576,
69
+ 0.13701310753822327,
70
+ 0.13330328464508057,
71
+ 1.7823294401168823,
72
+ 0.5954466462135315,
73
+ 0.6865788102149963
74
+ ],
75
+ "mean": [
76
+ 0.4040228037735655,
77
+ -0.05247618696989652,
78
+ 0.6993459771218912,
79
+ 0.07011111991957117,
80
+ -0.03520558904460197,
81
+ 0.14470242846953205,
82
+ 0.05491166606739672,
83
+ -0.19073058893378805,
84
+ -0.06988117661871228,
85
+ 0.021167300266524632,
86
+ 0.0018730097199051647,
87
+ -0.0015708771114029225,
88
+ -0.001553903484685181,
89
+ 0.014420973441752116,
90
+ 0.00961362082315661,
91
+ -0.0017634798849202702
92
+ ],
93
+ "std": [
94
+ 0.11861927913357073,
95
+ 0.11318943557281885,
96
+ 0.17849160646987516,
97
+ 0.4812528791886511,
98
+ 0.7487196258736938,
99
+ 0.4251126804030219,
100
+ 0.47797229807769137,
101
+ 0.6267695346407992,
102
+ 0.5782954306815554,
103
+ 0.01996590431050817,
104
+ 0.015762732601462,
105
+ 0.010587306860662312,
106
+ 0.01180823404609393,
107
+ 0.10086023299718701,
108
+ 0.04747329757360547,
109
+ 0.14801097710214692
110
+ ],
111
+ "count": [
112
+ 218367
113
+ ],
114
+ "q01": [
115
+ 0.23988881827972064,
116
+ -0.10387461883034811,
117
+ 0.6297243923476871,
118
+ -0.43622841955353464,
119
+ -0.6885486527617446,
120
+ -0.14103555388039019,
121
+ -0.4327210063987595,
122
+ -0.9304790915716145,
123
+ -0.3465462258490417,
124
+ -1.000000013351432e-10,
125
+ -0.02190519006673929,
126
+ -0.02092386152118027,
127
+ -0.026422736164554015,
128
+ -0.11497312558112925,
129
+ -0.025678747253255536,
130
+ -0.17124583807766966
131
+ ],
132
+ "q10": [
133
+ 0.3042306436755216,
134
+ -0.09609991620622624,
135
+ 0.6394063072781145,
136
+ -0.3351306352998669,
137
+ -0.5751963015002924,
138
+ -0.08490621283307866,
139
+ -0.33849701676224275,
140
+ -0.8340557712529717,
141
+ -0.29760290790884514,
142
+ -1.000000013351432e-10,
143
+ -0.013015102506385306,
144
+ -0.011160732577261764,
145
+ -0.012954131824952583,
146
+ -0.056165101267083004,
147
+ -0.01548487039073523,
148
+ -0.15972996251089885
149
+ ],
150
+ "q50": [
151
+ 0.4160225263528291,
152
+ -0.06267459152989589,
153
+ 0.6811777491060801,
154
+ 0.0543536902230377,
155
+ -0.14012251058819217,
156
+ 0.04772394280004727,
157
+ 0.03838119138716219,
158
+ -0.14865169094195635,
159
+ -0.2073069401172425,
160
+ 0.020772594934870346,
161
+ 0.0005058764599893245,
162
+ 4.561956601090609e-06,
163
+ 4.945756498042436e-05,
164
+ 0.0001528964582560528,
165
+ 0.00017815767540355586,
166
+ -0.0002718872707462707
167
+ ],
168
+ "q90": [
169
+ 0.46641008755679864,
170
+ 0.020233143101890064,
171
+ 0.8095135405259374,
172
+ 0.48172401592474906,
173
+ 0.7398737043070319,
174
+ 0.5970841215636613,
175
+ 0.4674752875999555,
176
+ 0.36281754777672676,
177
+ 0.5033361883834496,
178
+ 0.039998362213352394,
179
+ 0.016358020650084733,
180
+ 0.003987531854121596,
181
+ 0.006191431456332113,
182
+ 0.11275584021509974,
183
+ 0.06100667096893492,
184
+ 0.15292656959583054
185
+ ],
186
+ "q99": [
187
+ 0.47320723171849244,
188
+ 0.04114262476135705,
189
+ 0.8467967259270389,
190
+ 0.5867781409980988,
191
+ 0.8529584859257228,
192
+ 0.7953936054341789,
193
+ 0.5768269858732034,
194
+ 0.49353349323389695,
195
+ 0.7168914440776504,
196
+ 0.03999983541667277,
197
+ 0.03626038003704021,
198
+ 0.008814053033445103,
199
+ 0.012299866693624668,
200
+ 0.17731576759283033,
201
+ 0.12117986440120965,
202
+ 0.16474415715691368
203
+ ]
204
+ }
205
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/config.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "groot",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 16
9
+ ]
10
+ },
11
+ "observation.images.wrist": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 224,
16
+ 224
17
+ ]
18
+ },
19
+ "observation.images.right_shoulder": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 224,
24
+ 224
25
+ ]
26
+ },
27
+ "observation.images.guide": {
28
+ "type": "VISUAL",
29
+ "shape": [
30
+ 3,
31
+ 224,
32
+ 224
33
+ ]
34
+ }
35
+ },
36
+ "output_features": {
37
+ "action": {
38
+ "type": "ACTION",
39
+ "shape": [
40
+ 16
41
+ ]
42
+ }
43
+ },
44
+ "device": "cuda",
45
+ "use_amp": false,
46
+ "use_peft": false,
47
+ "push_to_hub": false,
48
+ "repo_id": null,
49
+ "private": null,
50
+ "tags": null,
51
+ "license": null,
52
+ "pretrained_path": null,
53
+ "pretrained_revision": null,
54
+ "chunk_size": 16,
55
+ "n_action_steps": 16,
56
+ "max_state_dim": 132,
57
+ "max_action_dim": 132,
58
+ "normalization_mapping": {
59
+ "VISUAL": "IDENTITY",
60
+ "STATE": "IDENTITY",
61
+ "ACTION": "IDENTITY"
62
+ },
63
+ "base_model_path": "nvidia/GR00T-N1.7-3B",
64
+ "action_decode_transform": null,
65
+ "embodiment_tag": "new_embodiment",
66
+ "tune_llm": false,
67
+ "tune_visual": false,
68
+ "tune_projector": true,
69
+ "tune_diffusion_model": true,
70
+ "tune_vlln": true,
71
+ "tune_top_llm_layers": 0,
72
+ "num_inference_timesteps": null,
73
+ "rtc_ramp_rate": null,
74
+ "use_flash_attention": false,
75
+ "use_relative_actions": false,
76
+ "relative_exclude_joints": [],
77
+ "optimizer_lr": 0.0001,
78
+ "optimizer_betas": [
79
+ 0.9,
80
+ 0.999
81
+ ],
82
+ "optimizer_eps": 1e-08,
83
+ "optimizer_weight_decay": 1e-05,
84
+ "warmup_ratio": 0.05,
85
+ "use_bf16": true,
86
+ "model_params_fp32": true,
87
+ "image_size": [
88
+ 256,
89
+ 256
90
+ ],
91
+ "tokenizer_assets_repo": null,
92
+ "lora_rank": 0,
93
+ "lora_alpha": 16,
94
+ "lora_dropout": 0.1,
95
+ "lora_full_model": false,
96
+ "video_backend": "decord",
97
+ "balance_dataset_weights": true,
98
+ "balance_trajectory_weights": true,
99
+ "dataset_paths": null,
100
+ "output_dir": "./tmp/gr00t",
101
+ "save_steps": 1000,
102
+ "max_steps": 10000,
103
+ "batch_size": 32,
104
+ "dataloader_num_workers": 8,
105
+ "report_to": "wandb",
106
+ "resume": false
107
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "groot_action_unpack_unnormalize_v2",
6
+ "config": {
7
+ "env_action_dim": 16,
8
+ "normalize_min_max": true,
9
+ "clip_normalized_action": true,
10
+ "libero_gripper_action": false,
11
+ "libero_gripper_binarize": true
12
+ },
13
+ "state_file": "policy_postprocessor_step_0_groot_action_unpack_unnormalize_v2.safetensors"
14
+ },
15
+ {
16
+ "registry_name": "device_processor",
17
+ "config": {
18
+ "device": "cpu",
19
+ "float_dtype": null
20
+ }
21
+ }
22
+ ]
23
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "groot_n1_7_pack_inputs_v1",
16
+ "config": {
17
+ "state_horizon": 1,
18
+ "action_horizon": 16,
19
+ "valid_action_horizon": 16,
20
+ "video_horizon": null,
21
+ "max_state_dim": 132,
22
+ "max_action_dim": 132,
23
+ "language_key": "task",
24
+ "formalize_language": true,
25
+ "embodiment_tag": "new_embodiment",
26
+ "embodiment_mapping": {
27
+ "oxe_droid_relative_eef_relative_joint": 24,
28
+ "xdof_relative_eef_relative_joint": 27,
29
+ "xdof_relative_eef_relative_joint_subtask": 27,
30
+ "real_g1_relative_eef_relative_joints": 25,
31
+ "real_r1_pro_sharpa_relative_eef": 26,
32
+ "real_r1_pro_sharpa_relative_eef_human": 26,
33
+ "real_r1_pro_sharpa_relative_eef_maxinsights": 26,
34
+ "real_r1_pro_sharpa_relative_eef_mecka": 26,
35
+ "unitree_g1_full_body_with_waist_height_nav_cmd": 25,
36
+ "simpler_env_google": 0,
37
+ "simpler_env_widowx": 1,
38
+ "libero_sim": 2,
39
+ "new_embodiment": 10
40
+ },
41
+ "normalize_min_max": true,
42
+ "state_dropout_prob": 0.0,
43
+ "clip_outliers": true,
44
+ "use_percentiles": false,
45
+ "video_modality_keys": null,
46
+ "raw_stats": null,
47
+ "modality_config": null
48
+ },
49
+ "state_file": "policy_preprocessor_step_2_groot_n1_7_pack_inputs_v1.safetensors"
50
+ },
51
+ {
52
+ "registry_name": "groot_n1_7_vlm_encode_v1",
53
+ "config": {
54
+ "model_name": "nvidia/Cosmos-Reason2-2B",
55
+ "image_crop_size": [
56
+ 230,
57
+ 230
58
+ ],
59
+ "image_target_size": [
60
+ 256,
61
+ 256
62
+ ],
63
+ "shortest_image_edge": null,
64
+ "crop_fraction": null,
65
+ "use_albumentations": false,
66
+ "letter_box_transform": false,
67
+ "device": "cuda"
68
+ }
69
+ },
70
+ {
71
+ "registry_name": "device_processor",
72
+ "config": {
73
+ "device": "cuda",
74
+ "float_dtype": null
75
+ }
76
+ }
77
+ ]
78
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/prompt_manifest.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "prompt_set": "detailed",
4
+ "source_file": "task_prompts.json",
5
+ "source_sha256": "0ba48fb372da9f49dcfcd7dcc2da8cce6ce991d3661060e664a104778d1ffacc",
6
+ "mapping_sha256": "f71954786e94cdeceb135dde09fdc6b65db0b87ab3b09d25060129d0f0fde018",
7
+ "dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
8
+ "dataset_revision": "v3.0",
9
+ "task_count": 13,
10
+ "task_codes": [
11
+ "5f",
12
+ "5g",
13
+ "1ext",
14
+ "3b",
15
+ "5b",
16
+ "3a",
17
+ "3c",
18
+ "5h",
19
+ "5d",
20
+ "3d",
21
+ "5e",
22
+ "5a",
23
+ "5c"
24
+ ],
25
+ "prompts": {
26
+ "5f": "Close the bottle in clockwise direction.",
27
+ "5g": "Grip the cap on the sides indicated by the 'squeeze' arrow and open the bottle in clockwise direction.",
28
+ "1ext": "Find the arrow guide and open the indicated drawer.",
29
+ "3b": "Open the door, rotate counter-clockwise and push.",
30
+ "5b": "Open the bottle in counter-clockwise direction.",
31
+ "3a": "Open the door, rotate clockwise and push.",
32
+ "3c": "Open the door, rotate clockwise and pull.",
33
+ "5h": "Close the bottle in counter-clockwise direction.",
34
+ "5d": "Close the bottle in clockwise direction.",
35
+ "3d": "Open the door, rotate counter-clockwise and pull.",
36
+ "5e": "Close the bottle in counter-clockwise direction.",
37
+ "5a": "Grip the cap on the sides indicated by the 'squeeze' arrow and open the bottle in counter-clockwise direction.",
38
+ "5c": "Open the bottle in clockwise direction."
39
+ }
40
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,258 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
4
+ "root": "/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2",
5
+ "episodes": null,
6
+ "image_transforms": {
7
+ "enable": false,
8
+ "max_num_transforms": 3,
9
+ "random_order": false,
10
+ "tfs": {
11
+ "brightness": {
12
+ "weight": 1.0,
13
+ "type": "ColorJitter",
14
+ "kwargs": {
15
+ "brightness": [
16
+ 0.8,
17
+ 1.2
18
+ ]
19
+ }
20
+ },
21
+ "contrast": {
22
+ "weight": 1.0,
23
+ "type": "ColorJitter",
24
+ "kwargs": {
25
+ "contrast": [
26
+ 0.8,
27
+ 1.2
28
+ ]
29
+ }
30
+ },
31
+ "saturation": {
32
+ "weight": 1.0,
33
+ "type": "ColorJitter",
34
+ "kwargs": {
35
+ "saturation": [
36
+ 0.5,
37
+ 1.5
38
+ ]
39
+ }
40
+ },
41
+ "hue": {
42
+ "weight": 1.0,
43
+ "type": "ColorJitter",
44
+ "kwargs": {
45
+ "hue": [
46
+ -0.05,
47
+ 0.05
48
+ ]
49
+ }
50
+ },
51
+ "sharpness": {
52
+ "weight": 1.0,
53
+ "type": "SharpnessJitter",
54
+ "kwargs": {
55
+ "sharpness": [
56
+ 0.5,
57
+ 1.5
58
+ ]
59
+ }
60
+ },
61
+ "affine": {
62
+ "weight": 1.0,
63
+ "type": "RandomAffine",
64
+ "kwargs": {
65
+ "degrees": [
66
+ -5.0,
67
+ 5.0
68
+ ],
69
+ "translate": [
70
+ 0.05,
71
+ 0.05
72
+ ]
73
+ }
74
+ }
75
+ }
76
+ },
77
+ "revision": null,
78
+ "use_imagenet_stats": true,
79
+ "video_backend": "torchcodec",
80
+ "return_uint8": false,
81
+ "depth_output_unit": "mm",
82
+ "streaming": false,
83
+ "eval_split": 0.0
84
+ },
85
+ "env": null,
86
+ "policy": {
87
+ "type": "groot",
88
+ "n_obs_steps": 1,
89
+ "input_features": {
90
+ "observation.state": {
91
+ "type": "STATE",
92
+ "shape": [
93
+ 16
94
+ ]
95
+ },
96
+ "observation.images.wrist": {
97
+ "type": "VISUAL",
98
+ "shape": [
99
+ 3,
100
+ 224,
101
+ 224
102
+ ]
103
+ },
104
+ "observation.images.right_shoulder": {
105
+ "type": "VISUAL",
106
+ "shape": [
107
+ 3,
108
+ 224,
109
+ 224
110
+ ]
111
+ },
112
+ "observation.images.guide": {
113
+ "type": "VISUAL",
114
+ "shape": [
115
+ 3,
116
+ 224,
117
+ 224
118
+ ]
119
+ }
120
+ },
121
+ "output_features": {
122
+ "action": {
123
+ "type": "ACTION",
124
+ "shape": [
125
+ 16
126
+ ]
127
+ }
128
+ },
129
+ "device": "cuda",
130
+ "use_amp": false,
131
+ "use_peft": false,
132
+ "push_to_hub": false,
133
+ "repo_id": null,
134
+ "private": null,
135
+ "tags": null,
136
+ "license": null,
137
+ "pretrained_path": null,
138
+ "pretrained_revision": null,
139
+ "chunk_size": 16,
140
+ "n_action_steps": 16,
141
+ "max_state_dim": 132,
142
+ "max_action_dim": 132,
143
+ "normalization_mapping": {
144
+ "VISUAL": "IDENTITY",
145
+ "STATE": "IDENTITY",
146
+ "ACTION": "IDENTITY"
147
+ },
148
+ "base_model_path": "nvidia/GR00T-N1.7-3B",
149
+ "action_decode_transform": null,
150
+ "embodiment_tag": "new_embodiment",
151
+ "tune_llm": false,
152
+ "tune_visual": false,
153
+ "tune_projector": true,
154
+ "tune_diffusion_model": true,
155
+ "tune_vlln": true,
156
+ "tune_top_llm_layers": 0,
157
+ "num_inference_timesteps": null,
158
+ "rtc_ramp_rate": null,
159
+ "use_flash_attention": false,
160
+ "use_relative_actions": false,
161
+ "relative_exclude_joints": [],
162
+ "optimizer_lr": 0.0001,
163
+ "optimizer_betas": [
164
+ 0.9,
165
+ 0.999
166
+ ],
167
+ "optimizer_eps": 1e-08,
168
+ "optimizer_weight_decay": 1e-05,
169
+ "warmup_ratio": 0.05,
170
+ "use_bf16": true,
171
+ "model_params_fp32": true,
172
+ "image_size": [
173
+ 256,
174
+ 256
175
+ ],
176
+ "tokenizer_assets_repo": null,
177
+ "lora_rank": 0,
178
+ "lora_alpha": 16,
179
+ "lora_dropout": 0.1,
180
+ "lora_full_model": false,
181
+ "video_backend": "decord",
182
+ "balance_dataset_weights": true,
183
+ "balance_trajectory_weights": true,
184
+ "dataset_paths": null,
185
+ "output_dir": "./tmp/gr00t",
186
+ "save_steps": 1000,
187
+ "max_steps": 10000,
188
+ "batch_size": 32,
189
+ "dataloader_num_workers": 8,
190
+ "report_to": "wandb",
191
+ "resume": false
192
+ },
193
+ "reward_model": null,
194
+ "output_dir": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline",
195
+ "job_name": "INSIGHT_RD_baseline",
196
+ "resume": false,
197
+ "seed": 42,
198
+ "cudnn_deterministic": false,
199
+ "num_workers": 4,
200
+ "batch_size": 128,
201
+ "prefetch_factor": 4,
202
+ "persistent_workers": true,
203
+ "steps": 60000,
204
+ "env_eval_freq": 20000,
205
+ "log_freq": 200,
206
+ "eval_steps": 0,
207
+ "max_eval_samples": 0,
208
+ "tolerance_s": 0.0001,
209
+ "save_checkpoint": true,
210
+ "save_freq": 20000,
211
+ "use_policy_training_preset": true,
212
+ "optimizer": {
213
+ "type": "adamw",
214
+ "lr": 0.0001,
215
+ "weight_decay": 1e-05,
216
+ "grad_clip_norm": 1.0,
217
+ "betas": [
218
+ 0.9,
219
+ 0.999
220
+ ],
221
+ "eps": 1e-08
222
+ },
223
+ "scheduler": {
224
+ "type": "diffuser",
225
+ "num_warmup_steps": 500,
226
+ "name": "cosine"
227
+ },
228
+ "eval": {
229
+ "n_episodes": 50,
230
+ "batch_size": 50,
231
+ "use_async_envs": true,
232
+ "recording": false,
233
+ "recording_repo_id": null,
234
+ "recording_private": false
235
+ },
236
+ "wandb": {
237
+ "enable": true,
238
+ "disable_artifact": true,
239
+ "project": "lerobot",
240
+ "entity": null,
241
+ "notes": null,
242
+ "run_id": "h160hqm2",
243
+ "mode": null,
244
+ "add_tags": true
245
+ },
246
+ "peft": null,
247
+ "job": {
248
+ "target": null,
249
+ "image": "huggingface/lerobot-gpu:latest",
250
+ "timeout": "2d",
251
+ "detach": false,
252
+ "tags": []
253
+ },
254
+ "save_checkpoint_to_hub": false,
255
+ "sample_weighting": null,
256
+ "rename_map": {},
257
+ "checkpoint_path": null
258
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/prompt_manifest.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "prompt_set": "detailed",
4
+ "source_file": "task_prompts.json",
5
+ "source_sha256": "0ba48fb372da9f49dcfcd7dcc2da8cce6ce991d3661060e664a104778d1ffacc",
6
+ "mapping_sha256": "f71954786e94cdeceb135dde09fdc6b65db0b87ab3b09d25060129d0f0fde018",
7
+ "dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
8
+ "dataset_revision": "v3.0",
9
+ "task_count": 13,
10
+ "task_codes": [
11
+ "5f",
12
+ "5g",
13
+ "1ext",
14
+ "3b",
15
+ "5b",
16
+ "3a",
17
+ "3c",
18
+ "5h",
19
+ "5d",
20
+ "3d",
21
+ "5e",
22
+ "5a",
23
+ "5c"
24
+ ],
25
+ "prompts": {
26
+ "5f": "Close the bottle in clockwise direction.",
27
+ "5g": "Grip the cap on the sides indicated by the 'squeeze' arrow and open the bottle in clockwise direction.",
28
+ "1ext": "Find the arrow guide and open the indicated drawer.",
29
+ "3b": "Open the door, rotate counter-clockwise and push.",
30
+ "5b": "Open the bottle in counter-clockwise direction.",
31
+ "3a": "Open the door, rotate clockwise and push.",
32
+ "3c": "Open the door, rotate clockwise and pull.",
33
+ "5h": "Close the bottle in counter-clockwise direction.",
34
+ "5d": "Close the bottle in clockwise direction.",
35
+ "3d": "Open the door, rotate counter-clockwise and pull.",
36
+ "5e": "Close the bottle in counter-clockwise direction.",
37
+ "5a": "Grip the cap on the sides indicated by the 'squeeze' arrow and open the bottle in counter-clockwise direction.",
38
+ "5c": "Open the bottle in clockwise direction."
39
+ }
40
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/optimizer_param_groups.json ADDED
@@ -0,0 +1,577 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "lr": 5.0659979290537954e-05,
4
+ "betas": [
5
+ 0.9,
6
+ 0.999
7
+ ],
8
+ "eps": 1e-08,
9
+ "weight_decay": 1e-05,
10
+ "amsgrad": false,
11
+ "maximize": false,
12
+ "foreach": null,
13
+ "capturable": false,
14
+ "differentiable": false,
15
+ "fused": null,
16
+ "decoupled_weight_decay": true,
17
+ "initial_lr": 0.0001,
18
+ "params": [
19
+ 0,
20
+ 1,
21
+ 2,
22
+ 3,
23
+ 4,
24
+ 5,
25
+ 6,
26
+ 7,
27
+ 8,
28
+ 9,
29
+ 10,
30
+ 11,
31
+ 12,
32
+ 13,
33
+ 14,
34
+ 15,
35
+ 16,
36
+ 17,
37
+ 18,
38
+ 19,
39
+ 20,
40
+ 21,
41
+ 22,
42
+ 23,
43
+ 24,
44
+ 25,
45
+ 26,
46
+ 27,
47
+ 28,
48
+ 29,
49
+ 30,
50
+ 31,
51
+ 32,
52
+ 33,
53
+ 34,
54
+ 35,
55
+ 36,
56
+ 37,
57
+ 38,
58
+ 39,
59
+ 40,
60
+ 41,
61
+ 42,
62
+ 43,
63
+ 44,
64
+ 45,
65
+ 46,
66
+ 47,
67
+ 48,
68
+ 49,
69
+ 50,
70
+ 51,
71
+ 52,
72
+ 53,
73
+ 54,
74
+ 55,
75
+ 56,
76
+ 57,
77
+ 58,
78
+ 59,
79
+ 60,
80
+ 61,
81
+ 62,
82
+ 63,
83
+ 64,
84
+ 65,
85
+ 66,
86
+ 67,
87
+ 68,
88
+ 69,
89
+ 70,
90
+ 71,
91
+ 72,
92
+ 73,
93
+ 74,
94
+ 75,
95
+ 76,
96
+ 77,
97
+ 78,
98
+ 79,
99
+ 80,
100
+ 81,
101
+ 82,
102
+ 83,
103
+ 84,
104
+ 85,
105
+ 86,
106
+ 87,
107
+ 88,
108
+ 89,
109
+ 90,
110
+ 91,
111
+ 92,
112
+ 93,
113
+ 94,
114
+ 95,
115
+ 96,
116
+ 97,
117
+ 98,
118
+ 99,
119
+ 100,
120
+ 101,
121
+ 102,
122
+ 103,
123
+ 104,
124
+ 105,
125
+ 106,
126
+ 107,
127
+ 108,
128
+ 109,
129
+ 110,
130
+ 111,
131
+ 112,
132
+ 113,
133
+ 114,
134
+ 115,
135
+ 116,
136
+ 117,
137
+ 118,
138
+ 119,
139
+ 120,
140
+ 121,
141
+ 122,
142
+ 123,
143
+ 124,
144
+ 125,
145
+ 126,
146
+ 127,
147
+ 128,
148
+ 129,
149
+ 130,
150
+ 131,
151
+ 132,
152
+ 133,
153
+ 134,
154
+ 135,
155
+ 136,
156
+ 137,
157
+ 138,
158
+ 139,
159
+ 140,
160
+ 141,
161
+ 142,
162
+ 143,
163
+ 144,
164
+ 145,
165
+ 146,
166
+ 147,
167
+ 148,
168
+ 149,
169
+ 150,
170
+ 151,
171
+ 152,
172
+ 153,
173
+ 154,
174
+ 155,
175
+ 156,
176
+ 157,
177
+ 158,
178
+ 159,
179
+ 160,
180
+ 161,
181
+ 162,
182
+ 163,
183
+ 164,
184
+ 165,
185
+ 166,
186
+ 167,
187
+ 168,
188
+ 169,
189
+ 170,
190
+ 171,
191
+ 172,
192
+ 173,
193
+ 174,
194
+ 175,
195
+ 176,
196
+ 177,
197
+ 178,
198
+ 179,
199
+ 180,
200
+ 181,
201
+ 182,
202
+ 183,
203
+ 184,
204
+ 185,
205
+ 186,
206
+ 187,
207
+ 188,
208
+ 189,
209
+ 190,
210
+ 191,
211
+ 192,
212
+ 193,
213
+ 194,
214
+ 195,
215
+ 196,
216
+ 197,
217
+ 198,
218
+ 199,
219
+ 200,
220
+ 201,
221
+ 202,
222
+ 203,
223
+ 204,
224
+ 205,
225
+ 206,
226
+ 207,
227
+ 208,
228
+ 209,
229
+ 210,
230
+ 211,
231
+ 212,
232
+ 213,
233
+ 214,
234
+ 215,
235
+ 216,
236
+ 217,
237
+ 218,
238
+ 219,
239
+ 220,
240
+ 221,
241
+ 222,
242
+ 223,
243
+ 224,
244
+ 225,
245
+ 226,
246
+ 227,
247
+ 228,
248
+ 229,
249
+ 230,
250
+ 231,
251
+ 232,
252
+ 233,
253
+ 234,
254
+ 235,
255
+ 236,
256
+ 237,
257
+ 238,
258
+ 239,
259
+ 240,
260
+ 241,
261
+ 242,
262
+ 243,
263
+ 244,
264
+ 245,
265
+ 246,
266
+ 247,
267
+ 248,
268
+ 249,
269
+ 250,
270
+ 251,
271
+ 252,
272
+ 253,
273
+ 254,
274
+ 255,
275
+ 256,
276
+ 257,
277
+ 258,
278
+ 259,
279
+ 260,
280
+ 261,
281
+ 262,
282
+ 263,
283
+ 264,
284
+ 265,
285
+ 266
286
+ ]
287
+ },
288
+ {
289
+ "weight_decay": 0.0,
290
+ "lr": 5.0659979290537954e-05,
291
+ "betas": [
292
+ 0.9,
293
+ 0.999
294
+ ],
295
+ "eps": 1e-08,
296
+ "amsgrad": false,
297
+ "maximize": false,
298
+ "foreach": null,
299
+ "capturable": false,
300
+ "differentiable": false,
301
+ "fused": null,
302
+ "decoupled_weight_decay": true,
303
+ "initial_lr": 0.0001,
304
+ "params": [
305
+ 267,
306
+ 268,
307
+ 269,
308
+ 270,
309
+ 271,
310
+ 272,
311
+ 273,
312
+ 274,
313
+ 275,
314
+ 276,
315
+ 277,
316
+ 278,
317
+ 279,
318
+ 280,
319
+ 281,
320
+ 282,
321
+ 283,
322
+ 284,
323
+ 285,
324
+ 286,
325
+ 287,
326
+ 288,
327
+ 289,
328
+ 290,
329
+ 291,
330
+ 292,
331
+ 293,
332
+ 294,
333
+ 295,
334
+ 296,
335
+ 297,
336
+ 298,
337
+ 299,
338
+ 300,
339
+ 301,
340
+ 302,
341
+ 303,
342
+ 304,
343
+ 305,
344
+ 306,
345
+ 307,
346
+ 308,
347
+ 309,
348
+ 310,
349
+ 311,
350
+ 312,
351
+ 313,
352
+ 314,
353
+ 315,
354
+ 316,
355
+ 317,
356
+ 318,
357
+ 319,
358
+ 320,
359
+ 321,
360
+ 322,
361
+ 323,
362
+ 324,
363
+ 325,
364
+ 326,
365
+ 327,
366
+ 328,
367
+ 329,
368
+ 330,
369
+ 331,
370
+ 332,
371
+ 333,
372
+ 334,
373
+ 335,
374
+ 336,
375
+ 337,
376
+ 338,
377
+ 339,
378
+ 340,
379
+ 341,
380
+ 342,
381
+ 343,
382
+ 344,
383
+ 345,
384
+ 346,
385
+ 347,
386
+ 348,
387
+ 349,
388
+ 350,
389
+ 351,
390
+ 352,
391
+ 353,
392
+ 354,
393
+ 355,
394
+ 356,
395
+ 357,
396
+ 358,
397
+ 359,
398
+ 360,
399
+ 361,
400
+ 362,
401
+ 363,
402
+ 364,
403
+ 365,
404
+ 366,
405
+ 367,
406
+ 368,
407
+ 369,
408
+ 370,
409
+ 371,
410
+ 372,
411
+ 373,
412
+ 374,
413
+ 375,
414
+ 376,
415
+ 377,
416
+ 378,
417
+ 379,
418
+ 380,
419
+ 381,
420
+ 382,
421
+ 383,
422
+ 384,
423
+ 385,
424
+ 386,
425
+ 387,
426
+ 388,
427
+ 389,
428
+ 390,
429
+ 391,
430
+ 392,
431
+ 393,
432
+ 394,
433
+ 395,
434
+ 396,
435
+ 397,
436
+ 398,
437
+ 399,
438
+ 400,
439
+ 401,
440
+ 402,
441
+ 403,
442
+ 404,
443
+ 405,
444
+ 406,
445
+ 407,
446
+ 408,
447
+ 409,
448
+ 410,
449
+ 411,
450
+ 412,
451
+ 413,
452
+ 414,
453
+ 415,
454
+ 416,
455
+ 417,
456
+ 418,
457
+ 419,
458
+ 420,
459
+ 421,
460
+ 422,
461
+ 423,
462
+ 424,
463
+ 425,
464
+ 426,
465
+ 427,
466
+ 428,
467
+ 429,
468
+ 430,
469
+ 431,
470
+ 432,
471
+ 433,
472
+ 434,
473
+ 435,
474
+ 436,
475
+ 437,
476
+ 438,
477
+ 439,
478
+ 440,
479
+ 441,
480
+ 442,
481
+ 443,
482
+ 444,
483
+ 445,
484
+ 446,
485
+ 447,
486
+ 448,
487
+ 449,
488
+ 450,
489
+ 451,
490
+ 452,
491
+ 453,
492
+ 454,
493
+ 455,
494
+ 456,
495
+ 457,
496
+ 458,
497
+ 459,
498
+ 460,
499
+ 461,
500
+ 462,
501
+ 463,
502
+ 464,
503
+ 465,
504
+ 466,
505
+ 467,
506
+ 468,
507
+ 469,
508
+ 470,
509
+ 471,
510
+ 472,
511
+ 473,
512
+ 474,
513
+ 475,
514
+ 476,
515
+ 477,
516
+ 478,
517
+ 479,
518
+ 480,
519
+ 481,
520
+ 482,
521
+ 483,
522
+ 484,
523
+ 485,
524
+ 486,
525
+ 487,
526
+ 488,
527
+ 489,
528
+ 490,
529
+ 491,
530
+ 492,
531
+ 493,
532
+ 494,
533
+ 495,
534
+ 496,
535
+ 497,
536
+ 498,
537
+ 499,
538
+ 500,
539
+ 501,
540
+ 502,
541
+ 503,
542
+ 504,
543
+ 505,
544
+ 506,
545
+ 507,
546
+ 508,
547
+ 509,
548
+ 510,
549
+ 511,
550
+ 512,
551
+ 513,
552
+ 514,
553
+ 515,
554
+ 516,
555
+ 517,
556
+ 518,
557
+ 519,
558
+ 520,
559
+ 521,
560
+ 522,
561
+ 523,
562
+ 524,
563
+ 525,
564
+ 526,
565
+ 527,
566
+ 528,
567
+ 529,
568
+ 530,
569
+ 531,
570
+ 532,
571
+ 533,
572
+ 534,
573
+ 535,
574
+ 536
575
+ ]
576
+ }
577
+ ]
Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/scheduler_state.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_lrs": [
3
+ 0.0001,
4
+ 0.0001
5
+ ],
6
+ "last_epoch": 30000,
7
+ "_step_count": 30001,
8
+ "_is_initial": false,
9
+ "_get_lr_called_within_step": false,
10
+ "_last_lr": [
11
+ 5.0659979290537954e-05,
12
+ 5.0659979290537954e-05
13
+ ],
14
+ "lr_lambdas": [
15
+ null,
16
+ null
17
+ ]
18
+ }
Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/training_step.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "step": 30000,
3
+ "num_processes": 1,
4
+ "batch_size": 128
5
+ }
Abs_6D + DeltaEE/Baseline/wandb/debug-internal.log.rerun_20260719_231636 ADDED
The diff for this file is too large to render. See raw diff
 
Abs_6D + DeltaEE/Baseline/wandb/debug.log.rerun_20260719_231636 ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
2
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Configure stats pid to 2591594
3
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug.log
5
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug-internal.log
6
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():766] calling init triggers
7
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():771] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'Whalswp/INSIGHTfixposV4_filtered_multispace_v2', 'root': '/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'return_uint8': False, 'depth_output_unit': 'mm', 'streaming': False, 'eval_split': 0.0}, 'env': None, 'policy': {'type': 'groot', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': None, 'pretrained_revision': None, 'chunk_size': 16, 'n_action_steps': 16, 'max_state_dim': 132, 'max_action_dim': 132, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'ACTION': <NormalizationMode.IDENTITY: 'IDENTITY'>}, 'base_model_path': 'nvidia/GR00T-N1.7-3B', 'action_decode_transform': None, 'embodiment_tag': 'new_embodiment', 'tune_llm': False, 'tune_visual': False, 'tune_projector': True, 'tune_diffusion_model': True, 'tune_vlln': True, 'tune_top_llm_layers': 0, 'num_inference_timesteps': None, 'rtc_ramp_rate': None, 'use_flash_attention': False, 'use_relative_actions': False, 'relative_exclude_joints': [], 'optimizer_lr': 0.0001, 'optimizer_betas': [0.9, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-05, 'warmup_ratio': 0.05, 'use_bf16': True, 'model_params_fp32': True, 'image_size': [256, 256], 'tokenizer_assets_repo': None, 'lora_rank': 0, 'lora_alpha': 16, 'lora_dropout': 0.1, 'lora_full_model': False, 'video_backend': 'decord', 'balance_dataset_weights': True, 'balance_trajectory_weights': True, 'dataset_paths': None, 'output_dir': './tmp/gr00t', 'save_steps': 1000, 'max_steps': 10000, 'batch_size': 32, 'dataloader_num_workers': 8, 'report_to': 'wandb', 'resume': False}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline', 'job_name': 'INSIGHT_RD_baseline', 'resume': False, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 128, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 60000, 'env_eval_freq': 20000, 'log_freq': 200, 'eval_steps': 0, 'max_eval_samples': 0, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 0.0001, 'weight_decay': 1e-05, 'grad_clip_norm': 1.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 500, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': True, 'recording': False, 'recording_repo_id': None, 'recording_private': False}, 'wandb': {'enable': True, 'disable_artifact': True, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None, 'add_tags': True}, 'peft': None, 'job': {'target': None, 'image': 'huggingface/lerobot-gpu:latest', 'timeout': '2d', 'detach': False, 'tags': []}, 'save_checkpoint_to_hub': False, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': None, '_wandb': {}}
9
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():814] starting backend
10
+ 2026-07-19 23:16:49,214 INFO MainThread:2591594 [wandb_init.py:init():829] sending inform_init request
11
+ 2026-07-19 23:16:49,747 INFO MainThread:2591594 [wandb_init.py:init():834] backend started and connected
12
+ 2026-07-19 23:16:49,749 INFO MainThread:2591594 [wandb_init.py:init():904] updated telemetry
13
+ 2026-07-19 23:16:49,755 INFO MainThread:2591594 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
14
+ 2026-07-19 23:16:50,147 INFO MainThread:2591594 [wandb_init.py:init():972] starting run threads in backend
15
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_console_start():2523] atexit reg
16
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2373] redirect: wrap_raw
17
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2442] Wrapping output streams.
18
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2465] Redirects installed.
19
+ 2026-07-19 23:16:50,259 INFO MainThread:2591594 [wandb_init.py:init():1010] run started, returning control to user process
Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/files/wandb-metadata.json ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-107-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-07-19T14:16:48.996554Z",
5
+ "args": [
6
+ "--config_path=/tmp/insight_lerobot_cfg_00a4crn2/baseline.yaml",
7
+ "--batch_size=128",
8
+ "--output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline"
9
+ ],
10
+ "program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
11
+ "git": {
12
+ "remote": "git@github.com:paragon7060/clvla.git",
13
+ "commit": "a9df08f7da2d8eb1fa7f28708b4eb6f6dc81b1f9"
14
+ },
15
+ "root": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline",
16
+ "host": "DGX-H200-01",
17
+ "executable": "/home/ext_minje/miniforge3/envs/lerobot060_groot/bin/python",
18
+ "cpu_count": 112,
19
+ "cpu_count_logical": 224,
20
+ "gpu": "NVIDIA H200",
21
+ "gpu_count": 8,
22
+ "disk": {
23
+ "/": {
24
+ "total": "1888425144320",
25
+ "used": "814094340096"
26
+ }
27
+ },
28
+ "memory": {
29
+ "total": "2164177309696"
30
+ },
31
+ "gpu_nvidia": [
32
+ {
33
+ "name": "NVIDIA H200",
34
+ "memoryTotal": "150754820096",
35
+ "cudaCores": 16896,
36
+ "architecture": "Hopper",
37
+ "uuid": "GPU-1e531a45-480e-3c1a-e9a0-2ba2cc27b0a3"
38
+ },
39
+ {
40
+ "name": "NVIDIA H200",
41
+ "memoryTotal": "150754820096",
42
+ "cudaCores": 16896,
43
+ "architecture": "Hopper",
44
+ "uuid": "GPU-5d3d5ae2-ffde-89d9-8818-dc19e11ad2fb"
45
+ },
46
+ {
47
+ "name": "NVIDIA H200",
48
+ "memoryTotal": "150754820096",
49
+ "cudaCores": 16896,
50
+ "architecture": "Hopper",
51
+ "uuid": "GPU-305c0dae-dcf5-d7bb-7dd3-5477d2194275"
52
+ },
53
+ {
54
+ "name": "NVIDIA H200",
55
+ "memoryTotal": "150754820096",
56
+ "cudaCores": 16896,
57
+ "architecture": "Hopper",
58
+ "uuid": "GPU-ed8b7920-8881-4e8f-3e88-be1bbecafa21"
59
+ },
60
+ {
61
+ "name": "NVIDIA H200",
62
+ "memoryTotal": "150754820096",
63
+ "cudaCores": 16896,
64
+ "architecture": "Hopper",
65
+ "uuid": "GPU-a91176db-55f2-b284-bbbf-56626596947d"
66
+ },
67
+ {
68
+ "name": "NVIDIA H200",
69
+ "memoryTotal": "150754820096",
70
+ "cudaCores": 16896,
71
+ "architecture": "Hopper",
72
+ "uuid": "GPU-922ff6d5-a948-205a-0426-dbee495fc3dd"
73
+ },
74
+ {
75
+ "name": "NVIDIA H200",
76
+ "memoryTotal": "150754820096",
77
+ "cudaCores": 16896,
78
+ "architecture": "Hopper",
79
+ "uuid": "GPU-1fde7c75-18fe-e3db-a4c1-85e67522f55f"
80
+ },
81
+ {
82
+ "name": "NVIDIA H200",
83
+ "memoryTotal": "150754820096",
84
+ "cudaCores": 16896,
85
+ "architecture": "Hopper",
86
+ "uuid": "GPU-a9b1c49b-d527-97d8-ce91-7e50bd02154c"
87
+ }
88
+ ],
89
+ "cudaVersion": "12.9",
90
+ "writerId": "oh1ufaq099yr0d3lbr8xeenwsom96wwz"
91
+ }
Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug.log ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
2
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Configure stats pid to 2591594
3
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug.log
5
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug-internal.log
6
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():766] calling init triggers
7
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():771] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'Whalswp/INSIGHTfixposV4_filtered_multispace_v2', 'root': '/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'return_uint8': False, 'depth_output_unit': 'mm', 'streaming': False, 'eval_split': 0.0}, 'env': None, 'policy': {'type': 'groot', 'n_obs_steps': 1, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': None, 'pretrained_revision': None, 'chunk_size': 16, 'n_action_steps': 16, 'max_state_dim': 132, 'max_action_dim': 132, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'ACTION': <NormalizationMode.IDENTITY: 'IDENTITY'>}, 'base_model_path': 'nvidia/GR00T-N1.7-3B', 'action_decode_transform': None, 'embodiment_tag': 'new_embodiment', 'tune_llm': False, 'tune_visual': False, 'tune_projector': True, 'tune_diffusion_model': True, 'tune_vlln': True, 'tune_top_llm_layers': 0, 'num_inference_timesteps': None, 'rtc_ramp_rate': None, 'use_flash_attention': False, 'use_relative_actions': False, 'relative_exclude_joints': [], 'optimizer_lr': 0.0001, 'optimizer_betas': [0.9, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-05, 'warmup_ratio': 0.05, 'use_bf16': True, 'model_params_fp32': True, 'image_size': [256, 256], 'tokenizer_assets_repo': None, 'lora_rank': 0, 'lora_alpha': 16, 'lora_dropout': 0.1, 'lora_full_model': False, 'video_backend': 'decord', 'balance_dataset_weights': True, 'balance_trajectory_weights': True, 'dataset_paths': None, 'output_dir': './tmp/gr00t', 'save_steps': 1000, 'max_steps': 10000, 'batch_size': 32, 'dataloader_num_workers': 8, 'report_to': 'wandb', 'resume': False}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/Baseline', 'job_name': 'INSIGHT_RD_baseline', 'resume': False, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 128, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 60000, 'env_eval_freq': 20000, 'log_freq': 200, 'eval_steps': 0, 'max_eval_samples': 0, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 0.0001, 'weight_decay': 1e-05, 'grad_clip_norm': 1.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 500, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': True, 'recording': False, 'recording_repo_id': None, 'recording_private': False}, 'wandb': {'enable': True, 'disable_artifact': True, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None, 'add_tags': True}, 'peft': None, 'job': {'target': None, 'image': 'huggingface/lerobot-gpu:latest', 'timeout': '2d', 'detach': False, 'tags': []}, 'save_checkpoint_to_hub': False, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': None, '_wandb': {}}
9
+ 2026-07-19 23:16:48,997 INFO MainThread:2591594 [wandb_init.py:init():814] starting backend
10
+ 2026-07-19 23:16:49,214 INFO MainThread:2591594 [wandb_init.py:init():829] sending inform_init request
11
+ 2026-07-19 23:16:49,747 INFO MainThread:2591594 [wandb_init.py:init():834] backend started and connected
12
+ 2026-07-19 23:16:49,749 INFO MainThread:2591594 [wandb_init.py:init():904] updated telemetry
13
+ 2026-07-19 23:16:49,755 INFO MainThread:2591594 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
14
+ 2026-07-19 23:16:50,147 INFO MainThread:2591594 [wandb_init.py:init():972] starting run threads in backend
15
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_console_start():2523] atexit reg
16
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2373] redirect: wrap_raw
17
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2442] Wrapping output streams.
18
+ 2026-07-19 23:16:50,256 INFO MainThread:2591594 [wandb_run.py:_redirect():2465] Redirects installed.
19
+ 2026-07-19 23:16:50,259 INFO MainThread:2591594 [wandb_init.py:init():1010] run started, returning control to user process
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/phase_schedule.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "step_interval": "[start_step, until_step)",
4
+ "total_steps": 60000,
5
+ "optimizer_lifecycle": "single_continuous_optimizer",
6
+ "scheduler_lifecycle": "single_continuous_scheduler",
7
+ "phases": [
8
+ {
9
+ "index": 0,
10
+ "name": "lap_rkd_linear_decay_6K",
11
+ "start_step": 0,
12
+ "until_step": 6000,
13
+ "policy": {
14
+ "rkd_enabled": true,
15
+ "rkd_fm_loss_weight": 1.0,
16
+ "rkd_loss_weight": 1.0,
17
+ "rkd_loss_weight_end": 0.0,
18
+ "rkd_loss_weight_schedule": "linear",
19
+ "rkd_distance_weight": 1.0,
20
+ "rkd_angle_weight": 2.0
21
+ },
22
+ "trainable": {
23
+ "tune_llm": false,
24
+ "tune_visual": false,
25
+ "tune_top_llm_layers": 0,
26
+ "tune_projector": true,
27
+ "tune_diffusion_model": true,
28
+ "tune_vlln": true
29
+ }
30
+ },
31
+ {
32
+ "index": 1,
33
+ "name": "fm_only",
34
+ "start_step": 6000,
35
+ "until_step": 60000,
36
+ "policy": {
37
+ "rkd_enabled": false,
38
+ "rkd_fm_loss_weight": 1.0,
39
+ "rkd_loss_weight": 0.0,
40
+ "rkd_loss_weight_end": null,
41
+ "rkd_loss_weight_schedule": "linear",
42
+ "rkd_distance_weight": 1.0,
43
+ "rkd_angle_weight": 2.0
44
+ },
45
+ "trainable": {
46
+ "tune_llm": false,
47
+ "tune_visual": false,
48
+ "tune_top_llm_layers": 0,
49
+ "tune_projector": true,
50
+ "tune_diffusion_model": true,
51
+ "tune_vlln": true
52
+ }
53
+ }
54
+ ]
55
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/action_space_manifest.json ADDED
@@ -0,0 +1,205 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 2,
3
+ "action_space": "ee_abs_rot6d+ee_delta",
4
+ "action_views": [
5
+ "ee_abs_rot6d",
6
+ "ee_delta"
7
+ ],
8
+ "source_feature": null,
9
+ "canonical_feature": "action",
10
+ "dtype": "float32",
11
+ "shape": [
12
+ 16
13
+ ],
14
+ "names": [
15
+ "ee_abs_rot6d.x",
16
+ "ee_abs_rot6d.y",
17
+ "ee_abs_rot6d.z",
18
+ "ee_abs_rot6d.r_col0_x",
19
+ "ee_abs_rot6d.r_col0_y",
20
+ "ee_abs_rot6d.r_col0_z",
21
+ "ee_abs_rot6d.r_col1_x",
22
+ "ee_abs_rot6d.r_col1_y",
23
+ "ee_abs_rot6d.r_col1_z",
24
+ "ee_abs_rot6d.gripper",
25
+ "ee_delta.delta_x",
26
+ "ee_delta.delta_y",
27
+ "ee_delta.delta_z",
28
+ "ee_delta.delta_axis_angle_x",
29
+ "ee_delta.delta_axis_angle_y",
30
+ "ee_delta.delta_axis_angle_z"
31
+ ],
32
+ "dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
33
+ "dataset_revision": "v3.0",
34
+ "dataset_total_frames": 218367,
35
+ "dataset_total_episodes": 4930,
36
+ "dataset_fps": 10,
37
+ "stats_sha256": "75b95092e5fd9a1b16aa6dc7a478883927d7c7cce76c26b56b290b675ae2311a",
38
+ "stats": {
39
+ "min": [
40
+ -0.21784618496894836,
41
+ -0.4474811851978302,
42
+ 0.1702737659215927,
43
+ -0.9999986290931702,
44
+ -0.9999991655349731,
45
+ -0.9999999403953552,
46
+ -0.9999990463256836,
47
+ -0.9999998211860657,
48
+ -0.9999988079071045,
49
+ 0.0,
50
+ -0.09827068448066711,
51
+ -0.13628354668617249,
52
+ -0.12191051989793777,
53
+ -2.431101083755493,
54
+ -0.3816019296646118,
55
+ -0.5094270706176758
56
+ ],
57
+ "max": [
58
+ 0.6283698678016663,
59
+ 0.4420371949672699,
60
+ 1.051027774810791,
61
+ 0.9999999403953552,
62
+ 0.9999995231628418,
63
+ 1.0,
64
+ 0.9999995231628418,
65
+ 1.0,
66
+ 0.9999996423721313,
67
+ 0.03999999910593033,
68
+ 0.11647262424230576,
69
+ 0.13701310753822327,
70
+ 0.13330328464508057,
71
+ 1.7823294401168823,
72
+ 0.5954466462135315,
73
+ 0.6865788102149963
74
+ ],
75
+ "mean": [
76
+ 0.4040228037735655,
77
+ -0.05247618696989652,
78
+ 0.6993459771218912,
79
+ 0.07011111991957117,
80
+ -0.03520558904460197,
81
+ 0.14470242846953205,
82
+ 0.05491166606739672,
83
+ -0.19073058893378805,
84
+ -0.06988117661871228,
85
+ 0.021167300266524632,
86
+ 0.0018730097199051647,
87
+ -0.0015708771114029225,
88
+ -0.001553903484685181,
89
+ 0.014420973441752116,
90
+ 0.00961362082315661,
91
+ -0.0017634798849202702
92
+ ],
93
+ "std": [
94
+ 0.11861927913357073,
95
+ 0.11318943557281885,
96
+ 0.17849160646987516,
97
+ 0.4812528791886511,
98
+ 0.7487196258736938,
99
+ 0.4251126804030219,
100
+ 0.47797229807769137,
101
+ 0.6267695346407992,
102
+ 0.5782954306815554,
103
+ 0.01996590431050817,
104
+ 0.015762732601462,
105
+ 0.010587306860662312,
106
+ 0.01180823404609393,
107
+ 0.10086023299718701,
108
+ 0.04747329757360547,
109
+ 0.14801097710214692
110
+ ],
111
+ "count": [
112
+ 218367
113
+ ],
114
+ "q01": [
115
+ 0.23988881827972064,
116
+ -0.10387461883034811,
117
+ 0.6297243923476871,
118
+ -0.43622841955353464,
119
+ -0.6885486527617446,
120
+ -0.14103555388039019,
121
+ -0.4327210063987595,
122
+ -0.9304790915716145,
123
+ -0.3465462258490417,
124
+ -1.000000013351432e-10,
125
+ -0.02190519006673929,
126
+ -0.02092386152118027,
127
+ -0.026422736164554015,
128
+ -0.11497312558112925,
129
+ -0.025678747253255536,
130
+ -0.17124583807766966
131
+ ],
132
+ "q10": [
133
+ 0.3042306436755216,
134
+ -0.09609991620622624,
135
+ 0.6394063072781145,
136
+ -0.3351306352998669,
137
+ -0.5751963015002924,
138
+ -0.08490621283307866,
139
+ -0.33849701676224275,
140
+ -0.8340557712529717,
141
+ -0.29760290790884514,
142
+ -1.000000013351432e-10,
143
+ -0.013015102506385306,
144
+ -0.011160732577261764,
145
+ -0.012954131824952583,
146
+ -0.056165101267083004,
147
+ -0.01548487039073523,
148
+ -0.15972996251089885
149
+ ],
150
+ "q50": [
151
+ 0.4160225263528291,
152
+ -0.06267459152989589,
153
+ 0.6811777491060801,
154
+ 0.0543536902230377,
155
+ -0.14012251058819217,
156
+ 0.04772394280004727,
157
+ 0.03838119138716219,
158
+ -0.14865169094195635,
159
+ -0.2073069401172425,
160
+ 0.020772594934870346,
161
+ 0.0005058764599893245,
162
+ 4.561956601090609e-06,
163
+ 4.945756498042436e-05,
164
+ 0.0001528964582560528,
165
+ 0.00017815767540355586,
166
+ -0.0002718872707462707
167
+ ],
168
+ "q90": [
169
+ 0.46641008755679864,
170
+ 0.020233143101890064,
171
+ 0.8095135405259374,
172
+ 0.48172401592474906,
173
+ 0.7398737043070319,
174
+ 0.5970841215636613,
175
+ 0.4674752875999555,
176
+ 0.36281754777672676,
177
+ 0.5033361883834496,
178
+ 0.039998362213352394,
179
+ 0.016358020650084733,
180
+ 0.003987531854121596,
181
+ 0.006191431456332113,
182
+ 0.11275584021509974,
183
+ 0.06100667096893492,
184
+ 0.15292656959583054
185
+ ],
186
+ "q99": [
187
+ 0.47320723171849244,
188
+ 0.04114262476135705,
189
+ 0.8467967259270389,
190
+ 0.5867781409980988,
191
+ 0.8529584859257228,
192
+ 0.7953936054341789,
193
+ 0.5768269858732034,
194
+ 0.49353349323389695,
195
+ 0.7168914440776504,
196
+ 0.03999983541667277,
197
+ 0.03626038003704021,
198
+ 0.008814053033445103,
199
+ 0.012299866693624668,
200
+ 0.17731576759283033,
201
+ 0.12117986440120965,
202
+ 0.16474415715691368
203
+ ]
204
+ }
205
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/config.json ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "groot_rkd",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 16
9
+ ]
10
+ },
11
+ "observation.images.wrist": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 224,
16
+ 224
17
+ ]
18
+ },
19
+ "observation.images.right_shoulder": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 224,
24
+ 224
25
+ ]
26
+ },
27
+ "observation.images.guide": {
28
+ "type": "VISUAL",
29
+ "shape": [
30
+ 3,
31
+ 224,
32
+ 224
33
+ ]
34
+ }
35
+ },
36
+ "output_features": {
37
+ "action": {
38
+ "type": "ACTION",
39
+ "shape": [
40
+ 16
41
+ ]
42
+ }
43
+ },
44
+ "device": "cuda",
45
+ "use_amp": false,
46
+ "use_peft": false,
47
+ "push_to_hub": false,
48
+ "repo_id": null,
49
+ "private": null,
50
+ "tags": null,
51
+ "license": null,
52
+ "pretrained_path": null,
53
+ "pretrained_revision": null,
54
+ "chunk_size": 16,
55
+ "n_action_steps": 16,
56
+ "max_state_dim": 132,
57
+ "max_action_dim": 132,
58
+ "normalization_mapping": {
59
+ "VISUAL": "IDENTITY",
60
+ "STATE": "IDENTITY",
61
+ "ACTION": "IDENTITY"
62
+ },
63
+ "base_model_path": "nvidia/GR00T-N1.7-3B",
64
+ "action_decode_transform": null,
65
+ "embodiment_tag": "new_embodiment",
66
+ "tune_llm": false,
67
+ "tune_visual": false,
68
+ "tune_projector": true,
69
+ "tune_diffusion_model": true,
70
+ "tune_vlln": true,
71
+ "tune_top_llm_layers": 0,
72
+ "num_inference_timesteps": null,
73
+ "rtc_ramp_rate": null,
74
+ "use_flash_attention": false,
75
+ "use_relative_actions": false,
76
+ "relative_exclude_joints": [],
77
+ "optimizer_lr": 0.0001,
78
+ "optimizer_betas": [
79
+ 0.9,
80
+ 0.999
81
+ ],
82
+ "optimizer_eps": 1e-08,
83
+ "optimizer_weight_decay": 1e-05,
84
+ "warmup_steps": 6000,
85
+ "warmup_ratio": 0.05,
86
+ "use_bf16": true,
87
+ "model_params_fp32": true,
88
+ "image_size": [
89
+ 256,
90
+ 256
91
+ ],
92
+ "tokenizer_assets_repo": null,
93
+ "lora_rank": 0,
94
+ "lora_alpha": 16,
95
+ "lora_dropout": 0.1,
96
+ "lora_full_model": false,
97
+ "video_backend": "decord",
98
+ "balance_dataset_weights": true,
99
+ "balance_trajectory_weights": true,
100
+ "dataset_paths": null,
101
+ "output_dir": "./tmp/gr00t",
102
+ "save_steps": 1000,
103
+ "max_steps": 10000,
104
+ "batch_size": 32,
105
+ "dataloader_num_workers": 8,
106
+ "report_to": "wandb",
107
+ "resume": false,
108
+ "rkd_enabled": true,
109
+ "rkd_vae_checkpoint": "/home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84",
110
+ "rkd_student_pooling": "attention",
111
+ "rkd_student_attention_hidden_dim": 512,
112
+ "rkd_student_attention_output_dim": 512,
113
+ "rkd_fm_loss_weight": 1.0,
114
+ "rkd_loss_weight": 1.0,
115
+ "rkd_distance_weight": 1.0,
116
+ "rkd_angle_weight": 2.0
117
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/phase_schedule.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "step_interval": "[start_step, until_step)",
4
+ "total_steps": 60000,
5
+ "optimizer_lifecycle": "single_continuous_optimizer",
6
+ "scheduler_lifecycle": "single_continuous_scheduler",
7
+ "phases": [
8
+ {
9
+ "index": 0,
10
+ "name": "lap_rkd_linear_decay_6K",
11
+ "start_step": 0,
12
+ "until_step": 6000,
13
+ "policy": {
14
+ "rkd_enabled": true,
15
+ "rkd_fm_loss_weight": 1.0,
16
+ "rkd_loss_weight": 1.0,
17
+ "rkd_loss_weight_end": 0.0,
18
+ "rkd_loss_weight_schedule": "linear",
19
+ "rkd_distance_weight": 1.0,
20
+ "rkd_angle_weight": 2.0
21
+ },
22
+ "trainable": {
23
+ "tune_llm": false,
24
+ "tune_visual": false,
25
+ "tune_top_llm_layers": 0,
26
+ "tune_projector": true,
27
+ "tune_diffusion_model": true,
28
+ "tune_vlln": true
29
+ }
30
+ },
31
+ {
32
+ "index": 1,
33
+ "name": "fm_only",
34
+ "start_step": 6000,
35
+ "until_step": 60000,
36
+ "policy": {
37
+ "rkd_enabled": false,
38
+ "rkd_fm_loss_weight": 1.0,
39
+ "rkd_loss_weight": 0.0,
40
+ "rkd_loss_weight_end": null,
41
+ "rkd_loss_weight_schedule": "linear",
42
+ "rkd_distance_weight": 1.0,
43
+ "rkd_angle_weight": 2.0
44
+ },
45
+ "trainable": {
46
+ "tune_llm": false,
47
+ "tune_visual": false,
48
+ "tune_top_llm_layers": 0,
49
+ "tune_projector": true,
50
+ "tune_diffusion_model": true,
51
+ "tune_vlln": true
52
+ }
53
+ }
54
+ ]
55
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "groot_action_unpack_unnormalize_v2",
6
+ "config": {
7
+ "env_action_dim": 16,
8
+ "normalize_min_max": true,
9
+ "clip_normalized_action": true,
10
+ "libero_gripper_action": false,
11
+ "libero_gripper_binarize": true
12
+ },
13
+ "state_file": "policy_postprocessor_step_0_groot_action_unpack_unnormalize_v2.safetensors"
14
+ },
15
+ {
16
+ "registry_name": "device_processor",
17
+ "config": {
18
+ "device": "cpu",
19
+ "float_dtype": null
20
+ }
21
+ }
22
+ ]
23
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "rkd_preserve_raw_action",
16
+ "config": {}
17
+ },
18
+ {
19
+ "registry_name": "groot_n1_7_pack_inputs_v1",
20
+ "config": {
21
+ "state_horizon": 1,
22
+ "action_horizon": 16,
23
+ "valid_action_horizon": 16,
24
+ "video_horizon": null,
25
+ "max_state_dim": 132,
26
+ "max_action_dim": 132,
27
+ "language_key": "task",
28
+ "formalize_language": true,
29
+ "embodiment_tag": "new_embodiment",
30
+ "embodiment_mapping": {
31
+ "oxe_droid_relative_eef_relative_joint": 24,
32
+ "xdof_relative_eef_relative_joint": 27,
33
+ "xdof_relative_eef_relative_joint_subtask": 27,
34
+ "real_g1_relative_eef_relative_joints": 25,
35
+ "real_r1_pro_sharpa_relative_eef": 26,
36
+ "real_r1_pro_sharpa_relative_eef_human": 26,
37
+ "real_r1_pro_sharpa_relative_eef_maxinsights": 26,
38
+ "real_r1_pro_sharpa_relative_eef_mecka": 26,
39
+ "unitree_g1_full_body_with_waist_height_nav_cmd": 25,
40
+ "simpler_env_google": 0,
41
+ "simpler_env_widowx": 1,
42
+ "libero_sim": 2,
43
+ "new_embodiment": 10
44
+ },
45
+ "normalize_min_max": true,
46
+ "state_dropout_prob": 0.0,
47
+ "clip_outliers": true,
48
+ "use_percentiles": false,
49
+ "video_modality_keys": null,
50
+ "raw_stats": null,
51
+ "modality_config": null
52
+ },
53
+ "state_file": "policy_preprocessor_step_3_groot_n1_7_pack_inputs_v1.safetensors"
54
+ },
55
+ {
56
+ "registry_name": "groot_n1_7_vlm_encode_v1",
57
+ "config": {
58
+ "model_name": "nvidia/Cosmos-Reason2-2B",
59
+ "image_crop_size": [
60
+ 230,
61
+ 230
62
+ ],
63
+ "image_target_size": [
64
+ 256,
65
+ 256
66
+ ],
67
+ "shortest_image_edge": null,
68
+ "crop_fraction": null,
69
+ "use_albumentations": false,
70
+ "letter_box_transform": false,
71
+ "device": "cuda"
72
+ }
73
+ },
74
+ {
75
+ "registry_name": "device_processor",
76
+ "config": {
77
+ "device": "cuda",
78
+ "float_dtype": null
79
+ }
80
+ }
81
+ ]
82
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,268 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
4
+ "root": "/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2",
5
+ "episodes": null,
6
+ "image_transforms": {
7
+ "enable": false,
8
+ "max_num_transforms": 3,
9
+ "random_order": false,
10
+ "tfs": {
11
+ "brightness": {
12
+ "weight": 1.0,
13
+ "type": "ColorJitter",
14
+ "kwargs": {
15
+ "brightness": [
16
+ 0.8,
17
+ 1.2
18
+ ]
19
+ }
20
+ },
21
+ "contrast": {
22
+ "weight": 1.0,
23
+ "type": "ColorJitter",
24
+ "kwargs": {
25
+ "contrast": [
26
+ 0.8,
27
+ 1.2
28
+ ]
29
+ }
30
+ },
31
+ "saturation": {
32
+ "weight": 1.0,
33
+ "type": "ColorJitter",
34
+ "kwargs": {
35
+ "saturation": [
36
+ 0.5,
37
+ 1.5
38
+ ]
39
+ }
40
+ },
41
+ "hue": {
42
+ "weight": 1.0,
43
+ "type": "ColorJitter",
44
+ "kwargs": {
45
+ "hue": [
46
+ -0.05,
47
+ 0.05
48
+ ]
49
+ }
50
+ },
51
+ "sharpness": {
52
+ "weight": 1.0,
53
+ "type": "SharpnessJitter",
54
+ "kwargs": {
55
+ "sharpness": [
56
+ 0.5,
57
+ 1.5
58
+ ]
59
+ }
60
+ },
61
+ "affine": {
62
+ "weight": 1.0,
63
+ "type": "RandomAffine",
64
+ "kwargs": {
65
+ "degrees": [
66
+ -5.0,
67
+ 5.0
68
+ ],
69
+ "translate": [
70
+ 0.05,
71
+ 0.05
72
+ ]
73
+ }
74
+ }
75
+ }
76
+ },
77
+ "revision": null,
78
+ "use_imagenet_stats": true,
79
+ "video_backend": "torchcodec",
80
+ "return_uint8": false,
81
+ "depth_output_unit": "mm",
82
+ "streaming": false,
83
+ "eval_split": 0.0
84
+ },
85
+ "env": null,
86
+ "policy": {
87
+ "type": "groot_rkd",
88
+ "n_obs_steps": 1,
89
+ "input_features": {
90
+ "observation.state": {
91
+ "type": "STATE",
92
+ "shape": [
93
+ 16
94
+ ]
95
+ },
96
+ "observation.images.wrist": {
97
+ "type": "VISUAL",
98
+ "shape": [
99
+ 3,
100
+ 224,
101
+ 224
102
+ ]
103
+ },
104
+ "observation.images.right_shoulder": {
105
+ "type": "VISUAL",
106
+ "shape": [
107
+ 3,
108
+ 224,
109
+ 224
110
+ ]
111
+ },
112
+ "observation.images.guide": {
113
+ "type": "VISUAL",
114
+ "shape": [
115
+ 3,
116
+ 224,
117
+ 224
118
+ ]
119
+ }
120
+ },
121
+ "output_features": {
122
+ "action": {
123
+ "type": "ACTION",
124
+ "shape": [
125
+ 16
126
+ ]
127
+ }
128
+ },
129
+ "device": "cuda",
130
+ "use_amp": false,
131
+ "use_peft": false,
132
+ "push_to_hub": false,
133
+ "repo_id": null,
134
+ "private": null,
135
+ "tags": null,
136
+ "license": null,
137
+ "pretrained_path": null,
138
+ "pretrained_revision": null,
139
+ "chunk_size": 16,
140
+ "n_action_steps": 16,
141
+ "max_state_dim": 132,
142
+ "max_action_dim": 132,
143
+ "normalization_mapping": {
144
+ "VISUAL": "IDENTITY",
145
+ "STATE": "IDENTITY",
146
+ "ACTION": "IDENTITY"
147
+ },
148
+ "base_model_path": "nvidia/GR00T-N1.7-3B",
149
+ "action_decode_transform": null,
150
+ "embodiment_tag": "new_embodiment",
151
+ "tune_llm": false,
152
+ "tune_visual": false,
153
+ "tune_projector": true,
154
+ "tune_diffusion_model": true,
155
+ "tune_vlln": true,
156
+ "tune_top_llm_layers": 0,
157
+ "num_inference_timesteps": null,
158
+ "rtc_ramp_rate": null,
159
+ "use_flash_attention": false,
160
+ "use_relative_actions": false,
161
+ "relative_exclude_joints": [],
162
+ "optimizer_lr": 0.0001,
163
+ "optimizer_betas": [
164
+ 0.9,
165
+ 0.999
166
+ ],
167
+ "optimizer_eps": 1e-08,
168
+ "optimizer_weight_decay": 1e-05,
169
+ "warmup_steps": 6000,
170
+ "warmup_ratio": 0.05,
171
+ "use_bf16": true,
172
+ "model_params_fp32": true,
173
+ "image_size": [
174
+ 256,
175
+ 256
176
+ ],
177
+ "tokenizer_assets_repo": null,
178
+ "lora_rank": 0,
179
+ "lora_alpha": 16,
180
+ "lora_dropout": 0.1,
181
+ "lora_full_model": false,
182
+ "video_backend": "decord",
183
+ "balance_dataset_weights": true,
184
+ "balance_trajectory_weights": true,
185
+ "dataset_paths": null,
186
+ "output_dir": "./tmp/gr00t",
187
+ "save_steps": 1000,
188
+ "max_steps": 10000,
189
+ "batch_size": 32,
190
+ "dataloader_num_workers": 8,
191
+ "report_to": "wandb",
192
+ "resume": false,
193
+ "rkd_enabled": true,
194
+ "rkd_vae_checkpoint": "/home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84",
195
+ "rkd_student_pooling": "attention",
196
+ "rkd_student_attention_hidden_dim": 512,
197
+ "rkd_student_attention_output_dim": 512,
198
+ "rkd_fm_loss_weight": 1.0,
199
+ "rkd_loss_weight": 1.0,
200
+ "rkd_distance_weight": 1.0,
201
+ "rkd_angle_weight": 2.0
202
+ },
203
+ "reward_model": null,
204
+ "output_dir": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/LAPstyle_linear_6K",
205
+ "job_name": "INSIGHT_RD_RKD_TimewarpVAE_LAPstyle_6K",
206
+ "resume": false,
207
+ "seed": 42,
208
+ "cudnn_deterministic": false,
209
+ "num_workers": 4,
210
+ "batch_size": 128,
211
+ "prefetch_factor": 4,
212
+ "persistent_workers": true,
213
+ "steps": 60000,
214
+ "env_eval_freq": 20000,
215
+ "log_freq": 200,
216
+ "eval_steps": 0,
217
+ "max_eval_samples": 0,
218
+ "tolerance_s": 0.0001,
219
+ "save_checkpoint": true,
220
+ "save_freq": 20000,
221
+ "use_policy_training_preset": true,
222
+ "optimizer": {
223
+ "type": "adamw",
224
+ "lr": 0.0001,
225
+ "weight_decay": 1e-05,
226
+ "grad_clip_norm": 1.0,
227
+ "betas": [
228
+ 0.9,
229
+ 0.999
230
+ ],
231
+ "eps": 1e-08
232
+ },
233
+ "scheduler": {
234
+ "type": "diffuser",
235
+ "num_warmup_steps": 6000,
236
+ "name": "cosine"
237
+ },
238
+ "eval": {
239
+ "n_episodes": 50,
240
+ "batch_size": 50,
241
+ "use_async_envs": true,
242
+ "recording": false,
243
+ "recording_repo_id": null,
244
+ "recording_private": false
245
+ },
246
+ "wandb": {
247
+ "enable": true,
248
+ "disable_artifact": true,
249
+ "project": "lerobot",
250
+ "entity": null,
251
+ "notes": null,
252
+ "run_id": "r3cz38bm",
253
+ "mode": null,
254
+ "add_tags": true
255
+ },
256
+ "peft": null,
257
+ "job": {
258
+ "target": null,
259
+ "image": "huggingface/lerobot-gpu:latest",
260
+ "timeout": "2d",
261
+ "detach": false,
262
+ "tags": []
263
+ },
264
+ "save_checkpoint_to_hub": false,
265
+ "sample_weighting": null,
266
+ "rename_map": {},
267
+ "checkpoint_path": null
268
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/source_config.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # INSIGHT GR00T N1.7 RKD training with LAP-style early RKD guidance.
2
+ # Common RKD values live in the top-level policy; phases contain overrides only.
3
+ insight:
4
+ # Composite policy action: ee_abs_rot6d 10D + ee_delta 6D (duplicate gripper removed)
5
+ action_space: ee_abs_rot6d+ee_delta
6
+ # Options: coarse | detailed
7
+ prompt_set: detailed
8
+ save_steps: ["60K"]
9
+ phases:
10
+ - name: lap_rkd_linear_decay_6K
11
+ until_step: 6K
12
+ policy:
13
+ # Inherits RKD=1.0 below and linearly decays it to zero.
14
+ rkd_loss_weight_end: 0.0
15
+
16
+ - name: fm_only
17
+ until_step: 60K
18
+ policy:
19
+ rkd_enabled: false
20
+ rkd_loss_weight: 0.0
21
+
22
+ dataset:
23
+ repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
24
+ root: ${HOME}/INSIGHTfixposV4_filtered_multispace_v2
25
+
26
+ policy:
27
+ type: groot_rkd
28
+ device: cuda
29
+ chunk_size: 16
30
+ n_action_steps: 16
31
+ push_to_hub: false
32
+ # LAP Stage-2 warmup scaled to this 60K run: exactly 6K updates.
33
+ warmup_steps: 6000
34
+
35
+ # VLM
36
+ tune_llm: false
37
+ tune_visual: false
38
+ tune_top_llm_layers: 0
39
+ # Action Expert
40
+ tune_projector: true
41
+ tune_diffusion_model: true
42
+ tune_vlln: true
43
+
44
+ # Frozen RD TimewarpVAE teacher: ee_abs_rot6d 10D + ee_delta 6D (gripper deduplicated).
45
+ rkd_enabled: true
46
+ rkd_vae_checkpoint: ${HOME}/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84
47
+ rkd_student_pooling: attention # attention | mean
48
+ rkd_student_attention_hidden_dim: 512
49
+ rkd_student_attention_output_dim: 512
50
+ rkd_fm_loss_weight: 1.0
51
+ rkd_loss_weight: 1.0
52
+ rkd_distance_weight: 1.0
53
+ rkd_angle_weight: 2.0
54
+
55
+ seed: 42
56
+ # Single GPU/process, so this is also the global batch size.
57
+ batch_size: 128
58
+ steps: 60000
59
+ log_freq: 200
60
+ output_dir: "${HOME}/groot_insight/train_ckpt/Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K"
61
+ job_name: INSIGHT_RD_RKD_TimewarpVAE_LAPstyle_6K
62
+
63
+ wandb:
64
+ enable: true
65
+ disable_artifact: true
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/optimizer_param_groups.json ADDED
@@ -0,0 +1,585 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "lr": 5.868240888334653e-05,
4
+ "betas": [
5
+ 0.9,
6
+ 0.999
7
+ ],
8
+ "eps": 1e-08,
9
+ "weight_decay": 1e-05,
10
+ "amsgrad": false,
11
+ "maximize": false,
12
+ "foreach": null,
13
+ "capturable": false,
14
+ "differentiable": false,
15
+ "fused": null,
16
+ "decoupled_weight_decay": true,
17
+ "initial_lr": 0.0001,
18
+ "params": [
19
+ 0,
20
+ 1,
21
+ 2,
22
+ 3,
23
+ 4,
24
+ 5,
25
+ 6,
26
+ 7,
27
+ 8,
28
+ 9,
29
+ 10,
30
+ 11,
31
+ 12,
32
+ 13,
33
+ 14,
34
+ 15,
35
+ 16,
36
+ 17,
37
+ 18,
38
+ 19,
39
+ 20,
40
+ 21,
41
+ 22,
42
+ 23,
43
+ 24,
44
+ 25,
45
+ 26,
46
+ 27,
47
+ 28,
48
+ 29,
49
+ 30,
50
+ 31,
51
+ 32,
52
+ 33,
53
+ 34,
54
+ 35,
55
+ 36,
56
+ 37,
57
+ 38,
58
+ 39,
59
+ 40,
60
+ 41,
61
+ 42,
62
+ 43,
63
+ 44,
64
+ 45,
65
+ 46,
66
+ 47,
67
+ 48,
68
+ 49,
69
+ 50,
70
+ 51,
71
+ 52,
72
+ 53,
73
+ 54,
74
+ 55,
75
+ 56,
76
+ 57,
77
+ 58,
78
+ 59,
79
+ 60,
80
+ 61,
81
+ 62,
82
+ 63,
83
+ 64,
84
+ 65,
85
+ 66,
86
+ 67,
87
+ 68,
88
+ 69,
89
+ 70,
90
+ 71,
91
+ 72,
92
+ 73,
93
+ 74,
94
+ 75,
95
+ 76,
96
+ 77,
97
+ 78,
98
+ 79,
99
+ 80,
100
+ 81,
101
+ 82,
102
+ 83,
103
+ 84,
104
+ 85,
105
+ 86,
106
+ 87,
107
+ 88,
108
+ 89,
109
+ 90,
110
+ 91,
111
+ 92,
112
+ 93,
113
+ 94,
114
+ 95,
115
+ 96,
116
+ 97,
117
+ 98,
118
+ 99,
119
+ 100,
120
+ 101,
121
+ 102,
122
+ 103,
123
+ 104,
124
+ 105,
125
+ 106,
126
+ 107,
127
+ 108,
128
+ 109,
129
+ 110,
130
+ 111,
131
+ 112,
132
+ 113,
133
+ 114,
134
+ 115,
135
+ 116,
136
+ 117,
137
+ 118,
138
+ 119,
139
+ 120,
140
+ 121,
141
+ 122,
142
+ 123,
143
+ 124,
144
+ 125,
145
+ 126,
146
+ 127,
147
+ 128,
148
+ 129,
149
+ 130,
150
+ 131,
151
+ 132,
152
+ 133,
153
+ 134,
154
+ 135,
155
+ 136,
156
+ 137,
157
+ 138,
158
+ 139,
159
+ 140,
160
+ 141,
161
+ 142,
162
+ 143,
163
+ 144,
164
+ 145,
165
+ 146,
166
+ 147,
167
+ 148,
168
+ 149,
169
+ 150,
170
+ 151,
171
+ 152,
172
+ 153,
173
+ 154,
174
+ 155,
175
+ 156,
176
+ 157,
177
+ 158,
178
+ 159,
179
+ 160,
180
+ 161,
181
+ 162,
182
+ 163,
183
+ 164,
184
+ 165,
185
+ 166,
186
+ 167,
187
+ 168,
188
+ 169,
189
+ 170,
190
+ 171,
191
+ 172,
192
+ 173,
193
+ 174,
194
+ 175,
195
+ 176,
196
+ 177,
197
+ 178,
198
+ 179,
199
+ 180,
200
+ 181,
201
+ 182,
202
+ 183,
203
+ 184,
204
+ 185,
205
+ 186,
206
+ 187,
207
+ 188,
208
+ 189,
209
+ 190,
210
+ 191,
211
+ 192,
212
+ 193,
213
+ 194,
214
+ 195,
215
+ 196,
216
+ 197,
217
+ 198,
218
+ 199,
219
+ 200,
220
+ 201,
221
+ 202,
222
+ 203,
223
+ 204,
224
+ 205,
225
+ 206,
226
+ 207,
227
+ 208,
228
+ 209,
229
+ 210,
230
+ 211,
231
+ 212,
232
+ 213,
233
+ 214,
234
+ 215,
235
+ 216,
236
+ 217,
237
+ 218,
238
+ 219,
239
+ 220,
240
+ 221,
241
+ 222,
242
+ 223,
243
+ 224,
244
+ 225,
245
+ 226,
246
+ 227,
247
+ 228,
248
+ 229,
249
+ 230,
250
+ 231,
251
+ 232,
252
+ 233,
253
+ 234,
254
+ 235,
255
+ 236,
256
+ 237,
257
+ 238,
258
+ 239,
259
+ 240,
260
+ 241,
261
+ 242,
262
+ 243,
263
+ 244,
264
+ 245,
265
+ 246,
266
+ 247,
267
+ 248,
268
+ 249,
269
+ 250,
270
+ 251,
271
+ 252,
272
+ 253,
273
+ 254,
274
+ 255,
275
+ 256,
276
+ 257,
277
+ 258,
278
+ 259,
279
+ 260,
280
+ 261,
281
+ 262,
282
+ 263,
283
+ 264,
284
+ 265,
285
+ 266,
286
+ 267,
287
+ 268,
288
+ 269,
289
+ 270
290
+ ]
291
+ },
292
+ {
293
+ "weight_decay": 0.0,
294
+ "lr": 5.868240888334653e-05,
295
+ "betas": [
296
+ 0.9,
297
+ 0.999
298
+ ],
299
+ "eps": 1e-08,
300
+ "amsgrad": false,
301
+ "maximize": false,
302
+ "foreach": null,
303
+ "capturable": false,
304
+ "differentiable": false,
305
+ "fused": null,
306
+ "decoupled_weight_decay": true,
307
+ "initial_lr": 0.0001,
308
+ "params": [
309
+ 271,
310
+ 272,
311
+ 273,
312
+ 274,
313
+ 275,
314
+ 276,
315
+ 277,
316
+ 278,
317
+ 279,
318
+ 280,
319
+ 281,
320
+ 282,
321
+ 283,
322
+ 284,
323
+ 285,
324
+ 286,
325
+ 287,
326
+ 288,
327
+ 289,
328
+ 290,
329
+ 291,
330
+ 292,
331
+ 293,
332
+ 294,
333
+ 295,
334
+ 296,
335
+ 297,
336
+ 298,
337
+ 299,
338
+ 300,
339
+ 301,
340
+ 302,
341
+ 303,
342
+ 304,
343
+ 305,
344
+ 306,
345
+ 307,
346
+ 308,
347
+ 309,
348
+ 310,
349
+ 311,
350
+ 312,
351
+ 313,
352
+ 314,
353
+ 315,
354
+ 316,
355
+ 317,
356
+ 318,
357
+ 319,
358
+ 320,
359
+ 321,
360
+ 322,
361
+ 323,
362
+ 324,
363
+ 325,
364
+ 326,
365
+ 327,
366
+ 328,
367
+ 329,
368
+ 330,
369
+ 331,
370
+ 332,
371
+ 333,
372
+ 334,
373
+ 335,
374
+ 336,
375
+ 337,
376
+ 338,
377
+ 339,
378
+ 340,
379
+ 341,
380
+ 342,
381
+ 343,
382
+ 344,
383
+ 345,
384
+ 346,
385
+ 347,
386
+ 348,
387
+ 349,
388
+ 350,
389
+ 351,
390
+ 352,
391
+ 353,
392
+ 354,
393
+ 355,
394
+ 356,
395
+ 357,
396
+ 358,
397
+ 359,
398
+ 360,
399
+ 361,
400
+ 362,
401
+ 363,
402
+ 364,
403
+ 365,
404
+ 366,
405
+ 367,
406
+ 368,
407
+ 369,
408
+ 370,
409
+ 371,
410
+ 372,
411
+ 373,
412
+ 374,
413
+ 375,
414
+ 376,
415
+ 377,
416
+ 378,
417
+ 379,
418
+ 380,
419
+ 381,
420
+ 382,
421
+ 383,
422
+ 384,
423
+ 385,
424
+ 386,
425
+ 387,
426
+ 388,
427
+ 389,
428
+ 390,
429
+ 391,
430
+ 392,
431
+ 393,
432
+ 394,
433
+ 395,
434
+ 396,
435
+ 397,
436
+ 398,
437
+ 399,
438
+ 400,
439
+ 401,
440
+ 402,
441
+ 403,
442
+ 404,
443
+ 405,
444
+ 406,
445
+ 407,
446
+ 408,
447
+ 409,
448
+ 410,
449
+ 411,
450
+ 412,
451
+ 413,
452
+ 414,
453
+ 415,
454
+ 416,
455
+ 417,
456
+ 418,
457
+ 419,
458
+ 420,
459
+ 421,
460
+ 422,
461
+ 423,
462
+ 424,
463
+ 425,
464
+ 426,
465
+ 427,
466
+ 428,
467
+ 429,
468
+ 430,
469
+ 431,
470
+ 432,
471
+ 433,
472
+ 434,
473
+ 435,
474
+ 436,
475
+ 437,
476
+ 438,
477
+ 439,
478
+ 440,
479
+ 441,
480
+ 442,
481
+ 443,
482
+ 444,
483
+ 445,
484
+ 446,
485
+ 447,
486
+ 448,
487
+ 449,
488
+ 450,
489
+ 451,
490
+ 452,
491
+ 453,
492
+ 454,
493
+ 455,
494
+ 456,
495
+ 457,
496
+ 458,
497
+ 459,
498
+ 460,
499
+ 461,
500
+ 462,
501
+ 463,
502
+ 464,
503
+ 465,
504
+ 466,
505
+ 467,
506
+ 468,
507
+ 469,
508
+ 470,
509
+ 471,
510
+ 472,
511
+ 473,
512
+ 474,
513
+ 475,
514
+ 476,
515
+ 477,
516
+ 478,
517
+ 479,
518
+ 480,
519
+ 481,
520
+ 482,
521
+ 483,
522
+ 484,
523
+ 485,
524
+ 486,
525
+ 487,
526
+ 488,
527
+ 489,
528
+ 490,
529
+ 491,
530
+ 492,
531
+ 493,
532
+ 494,
533
+ 495,
534
+ 496,
535
+ 497,
536
+ 498,
537
+ 499,
538
+ 500,
539
+ 501,
540
+ 502,
541
+ 503,
542
+ 504,
543
+ 505,
544
+ 506,
545
+ 507,
546
+ 508,
547
+ 509,
548
+ 510,
549
+ 511,
550
+ 512,
551
+ 513,
552
+ 514,
553
+ 515,
554
+ 516,
555
+ 517,
556
+ 518,
557
+ 519,
558
+ 520,
559
+ 521,
560
+ 522,
561
+ 523,
562
+ 524,
563
+ 525,
564
+ 526,
565
+ 527,
566
+ 528,
567
+ 529,
568
+ 530,
569
+ 531,
570
+ 532,
571
+ 533,
572
+ 534,
573
+ 535,
574
+ 536,
575
+ 537,
576
+ 538,
577
+ 539,
578
+ 540,
579
+ 541,
580
+ 542,
581
+ 543,
582
+ 544
583
+ ]
584
+ }
585
+ ]
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/scheduler_state.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_lrs": [
3
+ 0.0001,
4
+ 0.0001
5
+ ],
6
+ "last_epoch": 30000,
7
+ "_step_count": 30001,
8
+ "_is_initial": false,
9
+ "_get_lr_called_within_step": false,
10
+ "_last_lr": [
11
+ 5.868240888334653e-05,
12
+ 5.868240888334653e-05
13
+ ],
14
+ "lr_lambdas": [
15
+ null,
16
+ null
17
+ ]
18
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_175602-r3cz38bm/logs/debug-internal.log ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T17:56:02.450320507Z","level":"INFO","msg":"wandb-core"}
2
+ {"time":"2026-07-21T17:56:02.450589634Z","level":"INFO","msg":"stream: starting","core version":"0.27.2"}
3
+ {"time":"2026-07-21T17:56:02.839842689Z","level":"INFO","msg":"stream: created new stream","id":"r3cz38bm"}
4
+ {"time":"2026-07-21T17:56:02.839921121Z","level":"INFO","msg":"handler: started"}
5
+ {"time":"2026-07-21T17:56:02.840060232Z","level":"INFO","msg":"stream: started"}
6
+ {"time":"2026-07-21T17:56:02.840275995Z","level":"INFO","msg":"sender: started"}
7
+ {"time":"2026-07-21T17:56:02.840295265Z","level":"INFO","msg":"writer: started","stream_id":"r3cz38bm"}
8
+ {"time":"2026-07-21T17:56:03.42922439Z","level":"INFO","msg":"filestream: sending request","total_files":1}
9
+ {"time":"2026-07-21T17:56:04.033931674Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
10
+ {"time":"2026-07-21T17:56:05.900762911Z","level":"INFO","msg":"flowcontrol: backed up, offloading to disk","recordNumber":63}
11
+ {"time":"2026-07-21T17:56:05.904099709Z","level":"INFO","msg":"flowcontrol: unblocked","totalOffloaded":27}
12
+ {"time":"2026-07-21T17:56:07.506129177Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-07-21T17:56:07.506600851Z","level":"INFO","msg":"filestream: sending request","total_files":2,"console_offset":0,"console_lines":100,"uploaded_len":5,"complete":true,"exit_code":1}
14
+ {"time":"2026-07-21T17:56:07.943693767Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
15
+ {"time":"2026-07-21T17:56:07.943819242Z","level":"INFO","msg":"stream: finishing up"}
16
+ {"time":"2026-07-21T17:56:07.943828205Z","level":"INFO","msg":"handler: closed"}
17
+ {"time":"2026-07-21T17:56:07.943854908Z","level":"INFO","msg":"sender: closed"}
18
+ {"time":"2026-07-21T17:56:07.943857792Z","level":"INFO","msg":"stream: all finished"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_181953-a9vyia29/logs/debug-internal.log ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T18:19:53.504350753Z","level":"INFO","msg":"wandb-core"}
2
+ {"time":"2026-07-21T18:19:53.504575949Z","level":"INFO","msg":"stream: starting","core version":"0.27.2"}
3
+ {"time":"2026-07-21T18:19:53.968671816Z","level":"INFO","msg":"stream: created new stream","id":"a9vyia29"}
4
+ {"time":"2026-07-21T18:19:53.96879706Z","level":"INFO","msg":"handler: started"}
5
+ {"time":"2026-07-21T18:19:53.968952201Z","level":"INFO","msg":"stream: started"}
6
+ {"time":"2026-07-21T18:19:53.969570356Z","level":"INFO","msg":"sender: started"}
7
+ {"time":"2026-07-21T18:19:53.969654035Z","level":"INFO","msg":"writer: started","stream_id":"a9vyia29"}
8
+ {"time":"2026-07-21T18:19:54.421067335Z","level":"ERROR","msg":"runupserter: failed to init run","error":"no data but must resume"}
9
+ {"time":"2026-07-21T18:19:54.43535805Z","level":"ERROR","msg":"sender: uploadConfigFile: runhandle: run not yet initialized"}
10
+ {"time":"2026-07-21T18:19:54.435595622Z","level":"ERROR","msg":"runfiles: runhandle: run not yet initialized"}
11
+ {"time":"2026-07-21T18:19:54.435665726Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
12
+ {"time":"2026-07-21T18:19:54.435672834Z","level":"INFO","msg":"stream: finishing up"}
13
+ {"time":"2026-07-21T18:19:54.435678297Z","level":"INFO","msg":"handler: closed"}
14
+ {"time":"2026-07-21T18:19:54.435701433Z","level":"INFO","msg":"sender: closed"}
15
+ {"time":"2026-07-21T18:19:54.435704433Z","level":"INFO","msg":"stream: all finished"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/requirements.txt ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ pyarrow==25.0.0
2
+ importlib_metadata==9.0.0
3
+ wheel==0.47.0
4
+ cloudpickle==3.1.2
5
+ nvidia-cusolver-cu12==11.7.3.90
6
+ multiprocess==0.70.19
7
+ torch==2.11.0+cu128
8
+ nvidia-nvjitlink-cu12==12.8.93
9
+ fsspec==2026.2.0
10
+ pytz==2026.2
11
+ hf-xet==1.5.1
12
+ tokenizers==0.22.2
13
+ opencv-python-headless==4.13.0.92
14
+ cuda-pathfinder==1.2.2
15
+ frozenlist==1.8.0
16
+ smmap==5.0.3
17
+ httpcore==1.0.9
18
+ nvidia-nvshmem-cu12==3.4.5
19
+ nvidia-cuda-nvrtc-cu12==12.8.93
20
+ annotated-types==0.7.0
21
+ platformdirs==4.10.0
22
+ click==8.4.2
23
+ pyyaml-include==1.4.1
24
+ MarkupSafe==3.0.3
25
+ Jinja2==3.1.6
26
+ urllib3==2.7.0
27
+ nvidia-curand-cu12==10.3.9.90
28
+ certifi==2026.6.17
29
+ nvidia-cuda-runtime-cu12==12.8.90
30
+ torchcodec==0.11.1
31
+ toml==0.10.2
32
+ av==15.1.0
33
+ nvidia-cufft-cu12==11.3.3.83
34
+ attrs==26.1.0
35
+ lerobot==0.6.0
36
+ PyYAML==6.0.3
37
+ safetensors==0.8.0
38
+ diffusers==0.35.2
39
+ requests==2.34.2
40
+ nvidia-nvtx-cu12==12.8.90
41
+ charset-normalizer==3.4.9
42
+ networkx==3.6.1
43
+ cmake==4.1.3
44
+ pydantic_core==2.46.4
45
+ timm==1.0.28
46
+ dm-tree==0.1.10
47
+ nvidia-cufile-cu12==1.13.1.3
48
+ termcolor==3.3.0
49
+ multidict==6.7.1
50
+ filelock==3.29.0
51
+ tzdata==2026.3
52
+ aiohappyeyeballs==2.7.1
53
+ httpx==0.28.1
54
+ nvidia-nccl-cu12==2.28.9
55
+ Pygments==2.20.0
56
+ einops==0.8.2
57
+ nvidia-cusparse-cu12==12.5.8.93
58
+ torchvision==0.26.0+cu128
59
+ dill==0.4.1
60
+ GitPython==3.1.51
61
+ nvidia-cuda-cupti-cu12==12.8.90
62
+ decord==0.6.0
63
+ sympy==1.14.0
64
+ cuda-toolkit==12.8.1
65
+ python-dateutil==2.9.0.post0
66
+ shellingham==1.5.4
67
+ propcache==0.5.2
68
+ Farama-Notifications==0.0.6
69
+ pillow==12.2.0
70
+ cuda-bindings==12.9.4
71
+ huggingface_hub==1.23.0
72
+ pandas==2.3.3
73
+ typing-inspect==0.9.0
74
+ packaging==25.0
75
+ rich==15.0.0
76
+ nvidia-cusparselt-cu12==0.7.1
77
+ setuptools==78.1.0
78
+ wrapt==2.2.2
79
+ mypy_extensions==1.1.0
80
+ tqdm==4.68.4
81
+ draccus==0.10.0
82
+ aiosignal==1.4.0
83
+ mpmath==1.3.0
84
+ numpy==2.2.6
85
+ typing_extensions==4.15.0
86
+ psutil==7.2.2
87
+ mdurl==0.1.2
88
+ nvidia-cublas-cu12==12.8.4.1
89
+ absl-py==2.5.0
90
+ gitdb==4.0.12
91
+ transformers==5.5.4
92
+ accelerate==1.14.0
93
+ pydantic==2.13.4
94
+ pip==26.1.2
95
+ markdown-it-py==4.2.0
96
+ typer==0.26.8
97
+ triton==3.6.0
98
+ wandb==0.27.2
99
+ regex==2026.7.10
100
+ yarl==1.24.2
101
+ jsonlines==4.0.0
102
+ mergedeep==1.3.4
103
+ aiohttp==3.14.1
104
+ datasets==4.8.5
105
+ gymnasium==1.3.0
106
+ peft==0.19.1
107
+ zipp==4.1.0
108
+ idna==3.18
109
+ nvidia-cudnn-cu12==9.19.0.56
110
+ h11==0.16.0
111
+ sentry-sdk==2.65.0
112
+ annotated-doc==0.0.4
113
+ protobuf==7.35.1
114
+ typing-inspection==0.4.2
115
+ anyio==4.14.2
116
+ xxhash==3.8.1
117
+ six==1.17.0
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/wandb-metadata.json ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-107-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-07-19T14:16:52.871802Z",
5
+ "args": [
6
+ "--config_path=/tmp/insight_lerobot_cfg_xp9n_fwv/RSCLstyle_RKD_cosine_decay_60K.yaml",
7
+ "--batch_size=128",
8
+ "--output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K"
9
+ ],
10
+ "program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
11
+ "git": {
12
+ "remote": "git@github.com:paragon7060/clvla.git",
13
+ "commit": "a9df08f7da2d8eb1fa7f28708b4eb6f6dc81b1f9"
14
+ },
15
+ "root": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K",
16
+ "host": "DGX-H200-01",
17
+ "executable": "/home/ext_minje/miniforge3/envs/lerobot060_groot/bin/python",
18
+ "cpu_count": 112,
19
+ "cpu_count_logical": 224,
20
+ "gpu": "NVIDIA H200",
21
+ "gpu_count": 8,
22
+ "disk": {
23
+ "/": {
24
+ "total": "1888425144320",
25
+ "used": "814078631936"
26
+ }
27
+ },
28
+ "memory": {
29
+ "total": "2164177309696"
30
+ },
31
+ "gpu_nvidia": [
32
+ {
33
+ "name": "NVIDIA H200",
34
+ "memoryTotal": "150754820096",
35
+ "cudaCores": 16896,
36
+ "architecture": "Hopper",
37
+ "uuid": "GPU-1e531a45-480e-3c1a-e9a0-2ba2cc27b0a3"
38
+ },
39
+ {
40
+ "name": "NVIDIA H200",
41
+ "memoryTotal": "150754820096",
42
+ "cudaCores": 16896,
43
+ "architecture": "Hopper",
44
+ "uuid": "GPU-5d3d5ae2-ffde-89d9-8818-dc19e11ad2fb"
45
+ },
46
+ {
47
+ "name": "NVIDIA H200",
48
+ "memoryTotal": "150754820096",
49
+ "cudaCores": 16896,
50
+ "architecture": "Hopper",
51
+ "uuid": "GPU-305c0dae-dcf5-d7bb-7dd3-5477d2194275"
52
+ },
53
+ {
54
+ "name": "NVIDIA H200",
55
+ "memoryTotal": "150754820096",
56
+ "cudaCores": 16896,
57
+ "architecture": "Hopper",
58
+ "uuid": "GPU-ed8b7920-8881-4e8f-3e88-be1bbecafa21"
59
+ },
60
+ {
61
+ "name": "NVIDIA H200",
62
+ "memoryTotal": "150754820096",
63
+ "cudaCores": 16896,
64
+ "architecture": "Hopper",
65
+ "uuid": "GPU-a91176db-55f2-b284-bbbf-56626596947d"
66
+ },
67
+ {
68
+ "name": "NVIDIA H200",
69
+ "memoryTotal": "150754820096",
70
+ "cudaCores": 16896,
71
+ "architecture": "Hopper",
72
+ "uuid": "GPU-922ff6d5-a948-205a-0426-dbee495fc3dd"
73
+ },
74
+ {
75
+ "name": "NVIDIA H200",
76
+ "memoryTotal": "150754820096",
77
+ "cudaCores": 16896,
78
+ "architecture": "Hopper",
79
+ "uuid": "GPU-1fde7c75-18fe-e3db-a4c1-85e67522f55f"
80
+ },
81
+ {
82
+ "name": "NVIDIA H200",
83
+ "memoryTotal": "150754820096",
84
+ "cudaCores": 16896,
85
+ "architecture": "Hopper",
86
+ "uuid": "GPU-a9b1c49b-d527-97d8-ce91-7e50bd02154c"
87
+ }
88
+ ],
89
+ "cudaVersion": "12.9",
90
+ "writerId": "ffc0astd1y85im20l2d6i15t2xt71qa6"
91
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/config.yaml ADDED
@@ -0,0 +1,333 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.27.2
4
+ e:
5
+ e7cqx8h2t71p0rei3xmd9d75oaf2n9n8:
6
+ args:
7
+ - --config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json
8
+ - --resume=true
9
+ - --dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
10
+ - --output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
11
+ codePath: benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
12
+ codePathLocal: my_scripts/lerobot_train_yaml.py
13
+ cpu_count: 112
14
+ cpu_count_logical: 224
15
+ cudaVersion: "12.8"
16
+ disk:
17
+ /:
18
+ total: "1888425144320"
19
+ used: "1054655565824"
20
+ executable: /home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python
21
+ git:
22
+ commit: c1ff610ae4b02827f1c984cef8bbf28a7889bd6f
23
+ remote: git@github.com:paragon7060/clvla.git
24
+ gpu: NVIDIA H200
25
+ gpu_count: 2
26
+ gpu_nvidia:
27
+ - architecture: Hopper
28
+ cudaCores: 16896
29
+ memoryTotal: "150754820096"
30
+ name: NVIDIA H200
31
+ uuid: GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991
32
+ - architecture: Hopper
33
+ cudaCores: 16896
34
+ memoryTotal: "150754820096"
35
+ name: NVIDIA H200
36
+ uuid: GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc
37
+ host: ext_minje-box
38
+ memory:
39
+ total: "2164176568320"
40
+ os: Linux-6.8.0-55-generic-x86_64-with-glibc2.39
41
+ program: /home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
42
+ python: CPython 3.12.13
43
+ root: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
44
+ startedAt: "2026-07-21T17:56:02.220495Z"
45
+ writerId: e7cqx8h2t71p0rei3xmd9d75oaf2n9n8
46
+ m: []
47
+ python_version: 3.12.13
48
+ t:
49
+ "1":
50
+ - 1
51
+ - 11
52
+ - 41
53
+ - 49
54
+ - 51
55
+ - 71
56
+ - 83
57
+ - 98
58
+ "2":
59
+ - 1
60
+ - 11
61
+ - 41
62
+ - 49
63
+ - 51
64
+ - 71
65
+ - 83
66
+ - 98
67
+ "3":
68
+ - 5
69
+ - 13
70
+ - 14
71
+ - 15
72
+ - 16
73
+ - 41
74
+ - 62
75
+ "4": 3.12.13
76
+ "5": 0.27.2
77
+ "6": 5.5.4
78
+ "10":
79
+ - 21
80
+ "12": 0.27.2
81
+ "13": linux-x86_64
82
+ batch_size:
83
+ value: 128
84
+ checkpoint_path:
85
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000
86
+ cudnn_deterministic:
87
+ value: false
88
+ dataset:
89
+ value:
90
+ depth_output_unit: mm
91
+ episodes: null
92
+ eval_split: 0
93
+ image_transforms:
94
+ enable: false
95
+ max_num_transforms: 3
96
+ random_order: false
97
+ tfs:
98
+ affine:
99
+ kwargs:
100
+ degrees:
101
+ - -5
102
+ - 5
103
+ translate:
104
+ - 0.05
105
+ - 0.05
106
+ type: RandomAffine
107
+ weight: 1
108
+ brightness:
109
+ kwargs:
110
+ brightness:
111
+ - 0.8
112
+ - 1.2
113
+ type: ColorJitter
114
+ weight: 1
115
+ contrast:
116
+ kwargs:
117
+ contrast:
118
+ - 0.8
119
+ - 1.2
120
+ type: ColorJitter
121
+ weight: 1
122
+ hue:
123
+ kwargs:
124
+ hue:
125
+ - -0.05
126
+ - 0.05
127
+ type: ColorJitter
128
+ weight: 1
129
+ saturation:
130
+ kwargs:
131
+ saturation:
132
+ - 0.5
133
+ - 1.5
134
+ type: ColorJitter
135
+ weight: 1
136
+ sharpness:
137
+ kwargs:
138
+ sharpness:
139
+ - 0.5
140
+ - 1.5
141
+ type: SharpnessJitter
142
+ weight: 1
143
+ repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
144
+ return_uint8: false
145
+ revision: null
146
+ root: /home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
147
+ streaming: false
148
+ use_imagenet_stats: true
149
+ video_backend: torchcodec
150
+ env:
151
+ value: null
152
+ env_eval_freq:
153
+ value: 20000
154
+ eval:
155
+ value:
156
+ batch_size: 50
157
+ n_episodes: 50
158
+ recording: false
159
+ recording_private: false
160
+ recording_repo_id: null
161
+ use_async_envs: true
162
+ eval_steps:
163
+ value: 0
164
+ job:
165
+ value:
166
+ detach: false
167
+ image: huggingface/lerobot-gpu:latest
168
+ tags: []
169
+ target: null
170
+ timeout: 2d
171
+ job_name:
172
+ value: INSIGHT_RD_RKD_TimewarpVAE_RSCLstyle_C60
173
+ log_freq:
174
+ value: 200
175
+ max_eval_samples:
176
+ value: 0
177
+ num_workers:
178
+ value: 4
179
+ optimizer:
180
+ value:
181
+ betas:
182
+ - 0.9
183
+ - 0.999
184
+ eps: 1e-08
185
+ grad_clip_norm: 1
186
+ lr: 0.0001
187
+ type: adamw
188
+ weight_decay: 1e-05
189
+ output_dir:
190
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
191
+ peft:
192
+ value: null
193
+ persistent_workers:
194
+ value: true
195
+ policy:
196
+ value:
197
+ action_decode_transform: null
198
+ balance_dataset_weights: true
199
+ balance_trajectory_weights: true
200
+ base_model_path: nvidia/GR00T-N1.7-3B
201
+ batch_size: 32
202
+ chunk_size: 16
203
+ dataloader_num_workers: 8
204
+ dataset_paths: null
205
+ device: cuda
206
+ embodiment_tag: new_embodiment
207
+ image_size:
208
+ - 256
209
+ - 256
210
+ input_features:
211
+ observation.images.guide:
212
+ shape:
213
+ - 3
214
+ - 224
215
+ - 224
216
+ type: VISUAL
217
+ observation.images.right_shoulder:
218
+ shape:
219
+ - 3
220
+ - 224
221
+ - 224
222
+ type: VISUAL
223
+ observation.images.wrist:
224
+ shape:
225
+ - 3
226
+ - 224
227
+ - 224
228
+ type: VISUAL
229
+ observation.state:
230
+ shape:
231
+ - 16
232
+ type: STATE
233
+ license: null
234
+ lora_alpha: 16
235
+ lora_dropout: 0.1
236
+ lora_full_model: false
237
+ lora_rank: 0
238
+ max_action_dim: 132
239
+ max_state_dim: 132
240
+ max_steps: 10000
241
+ model_params_fp32: true
242
+ n_action_steps: 16
243
+ n_obs_steps: 1
244
+ normalization_mapping:
245
+ ACTION: IDENTITY
246
+ STATE: IDENTITY
247
+ VISUAL: IDENTITY
248
+ num_inference_timesteps: null
249
+ optimizer_betas:
250
+ - 0.9
251
+ - 0.999
252
+ optimizer_eps: 1e-08
253
+ optimizer_lr: 0.0001
254
+ optimizer_weight_decay: 1e-05
255
+ output_dir: ./tmp/gr00t
256
+ output_features:
257
+ action:
258
+ shape:
259
+ - 16
260
+ type: ACTION
261
+ pretrained_path: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
262
+ pretrained_revision: null
263
+ private: null
264
+ push_to_hub: false
265
+ relative_exclude_joints: []
266
+ repo_id: null
267
+ report_to: wandb
268
+ resume: false
269
+ rkd_angle_weight: 2
270
+ rkd_distance_weight: 1
271
+ rkd_enabled: true
272
+ rkd_fm_loss_weight: 1
273
+ rkd_loss_weight: 1
274
+ rkd_student_attention_hidden_dim: 512
275
+ rkd_student_attention_output_dim: 512
276
+ rkd_student_pooling: attention
277
+ rkd_vae_checkpoint: /home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84
278
+ rtc_ramp_rate: null
279
+ save_steps: 1000
280
+ tags: null
281
+ tokenizer_assets_repo: null
282
+ tune_diffusion_model: true
283
+ tune_llm: false
284
+ tune_projector: true
285
+ tune_top_llm_layers: 0
286
+ tune_visual: false
287
+ tune_vlln: true
288
+ type: groot_rkd
289
+ use_amp: false
290
+ use_bf16: true
291
+ use_flash_attention: false
292
+ use_peft: false
293
+ use_relative_actions: false
294
+ video_backend: decord
295
+ warmup_ratio: 0.05
296
+ warmup_steps: 3000
297
+ prefetch_factor:
298
+ value: 4
299
+ resume:
300
+ value: true
301
+ reward_model:
302
+ value: null
303
+ sample_weighting:
304
+ value: null
305
+ save_checkpoint:
306
+ value: true
307
+ save_checkpoint_to_hub:
308
+ value: false
309
+ save_freq:
310
+ value: 20000
311
+ scheduler:
312
+ value:
313
+ name: cosine
314
+ num_warmup_steps: 3000
315
+ type: diffuser
316
+ seed:
317
+ value: 42
318
+ steps:
319
+ value: 60000
320
+ tolerance_s:
321
+ value: 0.0001
322
+ use_policy_training_preset:
323
+ value: true
324
+ wandb:
325
+ value:
326
+ add_tags: true
327
+ disable_artifact: true
328
+ enable: true
329
+ entity: null
330
+ mode: null
331
+ notes: null
332
+ project: lerobot
333
+ run_id: bepgb8uu
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/requirements.txt ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ tzdata==2026.3
2
+ nvidia-cufile-cu12==1.13.1.3
3
+ wandb==0.27.2
4
+ six==1.17.0
5
+ peft==0.19.1
6
+ nvidia-curand-cu12==10.3.9.90
7
+ nvidia-cuda-runtime-cu12==12.8.90
8
+ datasets==4.8.5
9
+ attrs==26.1.0
10
+ cloudpickle==3.1.2
11
+ packaging==25.0
12
+ nvidia-cublas-cu12==12.8.4.1
13
+ cmake==4.1.3
14
+ psutil==7.2.2
15
+ pydantic==2.13.4
16
+ GitPython==3.1.51
17
+ pillow==12.2.0
18
+ pytz==2026.2
19
+ cuda-pathfinder==1.2.2
20
+ nvidia-cusparse-cu12==12.5.8.93
21
+ typing_extensions==4.15.0
22
+ urllib3==2.7.0
23
+ pydantic_core==2.46.4
24
+ draccus==0.10.0
25
+ mypy_extensions==1.1.0
26
+ dill==0.4.1
27
+ typer==0.26.8
28
+ decord==0.6.0
29
+ tqdm==4.68.4
30
+ triton==3.6.0
31
+ idna==3.18
32
+ requests==2.34.2
33
+ nvidia-cudnn-cu12==9.19.0.56
34
+ mdurl==0.1.2
35
+ shellingham==1.5.4
36
+ zipp==4.1.0
37
+ timm==1.0.28
38
+ torch==2.11.0+cu128
39
+ tokenizers==0.22.2
40
+ nvidia-cusparselt-cu12==0.7.1
41
+ nvidia-cufft-cu12==11.3.3.83
42
+ nvidia-cusolver-cu12==11.7.3.90
43
+ opencv-python-headless==4.13.0.92
44
+ httpx==0.28.1
45
+ Jinja2==3.1.6
46
+ einops==0.8.2
47
+ mergedeep==1.3.4
48
+ platformdirs==4.10.0
49
+ nvidia-nvjitlink-cu12==12.8.93
50
+ lerobot==0.6.0
51
+ PyYAML==6.0.3
52
+ accelerate==1.14.0
53
+ anyio==4.14.2
54
+ gymnasium==1.3.0
55
+ pip==26.1.2
56
+ huggingface_hub==1.23.0
57
+ diffusers==0.35.2
58
+ annotated-doc==0.0.4
59
+ pyarrow==25.0.0
60
+ python-dateutil==2.9.0.post0
61
+ termcolor==3.3.0
62
+ nvidia-nccl-cu12==2.28.9
63
+ sentry-sdk==2.65.0
64
+ torchcodec==0.11.1
65
+ pyyaml-include==1.4.1
66
+ multiprocess==0.70.19
67
+ wrapt==2.2.2
68
+ typing-inspection==0.4.2
69
+ nvidia-cuda-nvrtc-cu12==12.8.93
70
+ Pygments==2.20.0
71
+ nvidia-cuda-cupti-cu12==12.8.90
72
+ certifi==2026.6.17
73
+ setuptools==78.1.0
74
+ charset-normalizer==3.4.9
75
+ propcache==0.5.2
76
+ multidict==6.7.1
77
+ regex==2026.7.10
78
+ frozenlist==1.8.0
79
+ smmap==5.0.3
80
+ hf-xet==1.5.1
81
+ torchvision==0.26.0+cu128
82
+ gitdb==4.0.12
83
+ transformers==5.5.4
84
+ rich==15.0.0
85
+ cuda-bindings==12.9.4
86
+ protobuf==7.35.1
87
+ aiosignal==1.4.0
88
+ jsonlines==4.0.0
89
+ yarl==1.24.2
90
+ filelock==3.29.0
91
+ av==15.1.0
92
+ networkx==3.6.1
93
+ sympy==1.14.0
94
+ wheel==0.47.0
95
+ nvidia-nvtx-cu12==12.8.90
96
+ xxhash==3.8.1
97
+ toml==0.10.2
98
+ MarkupSafe==3.0.3
99
+ pandas==2.3.3
100
+ safetensors==0.8.0
101
+ click==8.4.2
102
+ fsspec==2026.2.0
103
+ dm-tree==0.1.10
104
+ absl-py==2.5.0
105
+ Farama-Notifications==0.0.6
106
+ mpmath==1.3.0
107
+ importlib_metadata==9.0.0
108
+ markdown-it-py==4.2.0
109
+ h11==0.16.0
110
+ typing-inspect==0.9.0
111
+ httpcore==1.0.9
112
+ annotated-types==0.7.0
113
+ numpy==2.2.6
114
+ nvidia-nvshmem-cu12==3.4.5
115
+ aiohappyeyeballs==2.7.1
116
+ cuda-toolkit==12.8.1
117
+ aiohttp==3.14.1
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-metadata.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-55-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-07-21T17:56:02.220495Z",
5
+ "args": [
6
+ "--config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json",
7
+ "--resume=true",
8
+ "--dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2",
9
+ "--output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K"
10
+ ],
11
+ "program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
12
+ "codePath": "benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
13
+ "codePathLocal": "my_scripts/lerobot_train_yaml.py",
14
+ "git": {
15
+ "remote": "git@github.com:paragon7060/clvla.git",
16
+ "commit": "c1ff610ae4b02827f1c984cef8bbf28a7889bd6f"
17
+ },
18
+ "root": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K",
19
+ "host": "ext_minje-box",
20
+ "executable": "/home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python",
21
+ "cpu_count": 112,
22
+ "cpu_count_logical": 224,
23
+ "gpu": "NVIDIA H200",
24
+ "gpu_count": 2,
25
+ "disk": {
26
+ "/": {
27
+ "total": "1888425144320",
28
+ "used": "1054655565824"
29
+ }
30
+ },
31
+ "memory": {
32
+ "total": "2164176568320"
33
+ },
34
+ "gpu_nvidia": [
35
+ {
36
+ "name": "NVIDIA H200",
37
+ "memoryTotal": "150754820096",
38
+ "cudaCores": 16896,
39
+ "architecture": "Hopper",
40
+ "uuid": "GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991"
41
+ },
42
+ {
43
+ "name": "NVIDIA H200",
44
+ "memoryTotal": "150754820096",
45
+ "cudaCores": 16896,
46
+ "architecture": "Hopper",
47
+ "uuid": "GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc"
48
+ }
49
+ ],
50
+ "cudaVersion": "12.8",
51
+ "writerId": "e7cqx8h2t71p0rei3xmd9d75oaf2n9n8"
52
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/loss":0.0115696769207716,"train/rkd_teacher_latent_norm":1,"train/rkd_loss":0.0015607024542987349,"_runtime":120015,"_timestamp":1.7845906275011618e+09,"train/weighted_flow_matching_loss":0.011165226809680462,"_wandb":{"runtime":120015},"train/rkd_fm_loss_weight":1,"train/episodes":114436.63190866752,"_step":39600,"train/rkd_distance_loss":0.0003176771861035377,"train/training_phase_index":0,"train/gpu_mem_gb":52.89515282392502,"train/weighted_rkd_loss":0.0004044499364681542,"train/dataloading_s":1.5435121013406023,"train/rkd_loss_weight":0.2591461049345566,"train/samples":5068800,"train/rkd_runtime_enabled":1,"train/flow_matching_loss":0.011165226809680462,"train/steps":39600,"train/lr":2.8660040166848404e-05,"train/training_phase_start_step":0,"train/epochs":23.212298561595844,"train/rkd_angle_loss":0.0006215126486495137,"train/training_phase_trainable_params":1.622090881e+09,"train/update_s":1.601553964055929,"train/training_phase_until_step":60000,"train/samples_per_s":40.698668116487305,"train/grad_norm":0.11059212811291218,"train/rkd_student_feature_norm":11.796961784362791,"train/rkd_student_pooling_id":1}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-core.log ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T17:56:02.293409893Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpfixkvv42/port-689.txt","pid":689,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-07-21T17:56:02.293408752Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmppku0dr88/port-694.txt","pid":694,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
3
+ {"time":"2026-07-21T17:56:02.293419796Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpwlkoj7ru/port-699.txt","pid":699,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
4
+ {"time":"2026-07-21T17:56:02.293965788Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":694}
5
+ {"time":"2026-07-21T17:56:02.293974724Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-689-1300-1501970219/socket","Net":"unix"}}
6
+ {"time":"2026-07-21T17:56:02.293957121Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-694-1302-2065636693/socket","Net":"unix"}}
7
+ {"time":"2026-07-21T17:56:02.294073117Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":689}
8
+ {"time":"2026-07-21T17:56:02.294102115Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":699}
9
+ {"time":"2026-07-21T17:56:02.294099717Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-699-1301-1209873691/socket","Net":"unix"}}
10
+ {"time":"2026-07-21T17:56:02.443641956Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
11
+ {"time":"2026-07-21T17:56:02.443692689Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
12
+ {"time":"2026-07-21T17:56:02.443723955Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
13
+ {"time":"2026-07-21T17:56:02.449359295Z","level":"INFO","msg":"handleInformInit: received","streamId":"ny0tx2de","id":"1(@)"}
14
+ {"time":"2026-07-21T17:56:02.449465549Z","level":"INFO","msg":"handleInformInit: received","streamId":"bepgb8uu","id":"1(@)"}
15
+ {"time":"2026-07-21T17:56:02.450212786Z","level":"INFO","msg":"handleInformInit: received","streamId":"r3cz38bm","id":"1(@)"}
16
+ {"time":"2026-07-21T17:56:02.840064538Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"r3cz38bm","id":"1(@)"}
17
+ {"time":"2026-07-21T17:56:02.91536888Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"bepgb8uu","id":"1(@)"}
18
+ {"time":"2026-07-21T17:56:02.919390334Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"ny0tx2de","id":"1(@)"}
19
+ {"time":"2026-07-21T17:56:05.90087574Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
20
+ {"time":"2026-07-21T17:56:05.900929416Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
21
+ {"time":"2026-07-21T17:56:05.901000596Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
22
+ {"time":"2026-07-21T17:56:05.901006185Z","level":"INFO","msg":"server is shutting down"}
23
+ {"time":"2026-07-21T17:56:05.901069468Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
24
+ {"time":"2026-07-21T17:56:05.901068443Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-689-1300-1501970219/socket","Net":"unix"}}
25
+ {"time":"2026-07-21T17:56:06.032160078Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
26
+ {"time":"2026-07-21T17:56:06.032296572Z","level":"INFO","msg":"server is shutting down"}
27
+ {"time":"2026-07-21T17:56:06.032372501Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
28
+ {"time":"2026-07-21T17:56:06.03236437Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-699-1301-1209873691/socket","Net":"unix"}}
29
+ {"time":"2026-07-21T17:56:06.032413194Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
30
+ {"time":"2026-07-21T17:56:06.032484365Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
31
+ {"time":"2026-07-21T17:56:06.045496378Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
32
+ {"time":"2026-07-21T17:56:06.045562827Z","level":"INFO","msg":"server is shutting down"}
33
+ {"time":"2026-07-21T17:56:06.045564528Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
34
+ {"time":"2026-07-21T17:56:06.045617631Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
35
+ {"time":"2026-07-21T17:56:06.045624383Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
36
+ {"time":"2026-07-21T17:56:06.045745175Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-694-1302-2065636693/socket","Net":"unix"}}
37
+ {"time":"2026-07-21T17:56:07.944258135Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
38
+ {"time":"2026-07-21T17:56:07.944298625Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
39
+ {"time":"2026-07-21T17:56:07.944310343Z","level":"INFO","msg":"server is closed"}
40
+ {"time":"2026-07-21T17:56:08.055341261Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
41
+ {"time":"2026-07-21T17:56:08.055381916Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
42
+ {"time":"2026-07-21T17:56:08.055392579Z","level":"INFO","msg":"server is closed"}
43
+ {"time":"2026-07-21T17:56:08.143517208Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
44
+ {"time":"2026-07-21T17:56:08.143548949Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
45
+ {"time":"2026-07-21T17:56:08.143560973Z","level":"INFO","msg":"server is closed"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-internal.log ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T17:56:02.449589454Z","level":"INFO","msg":"wandb-core"}
2
+ {"time":"2026-07-21T17:56:02.450458617Z","level":"INFO","msg":"stream: starting","core version":"0.27.2"}
3
+ {"time":"2026-07-21T17:56:02.915208542Z","level":"INFO","msg":"stream: created new stream","id":"bepgb8uu"}
4
+ {"time":"2026-07-21T17:56:02.915365243Z","level":"INFO","msg":"stream: started"}
5
+ {"time":"2026-07-21T17:56:02.915442997Z","level":"INFO","msg":"writer: started","stream_id":"bepgb8uu"}
6
+ {"time":"2026-07-21T17:56:02.915487492Z","level":"INFO","msg":"handler: started"}
7
+ {"time":"2026-07-21T17:56:02.915497513Z","level":"INFO","msg":"sender: started"}
8
+ {"time":"2026-07-21T17:56:03.830353213Z","level":"INFO","msg":"filestream: sending request","total_files":1}
9
+ {"time":"2026-07-21T17:56:04.39597572Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
10
+ {"time":"2026-07-21T17:56:06.032108165Z","level":"INFO","msg":"flowcontrol: backed up, offloading to disk","recordNumber":63}
11
+ {"time":"2026-07-21T17:56:06.035552908Z","level":"INFO","msg":"flowcontrol: unblocked","totalOffloaded":27}
12
+ {"time":"2026-07-21T17:56:07.690611502Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
13
+ {"time":"2026-07-21T17:56:07.69097215Z","level":"INFO","msg":"filestream: sending request","total_files":2,"console_offset":0,"console_lines":100,"uploaded_len":5,"complete":true,"exit_code":1}
14
+ {"time":"2026-07-21T17:56:08.142999873Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
15
+ {"time":"2026-07-21T17:56:08.14312926Z","level":"INFO","msg":"stream: finishing up"}
16
+ {"time":"2026-07-21T17:56:08.143138724Z","level":"INFO","msg":"handler: closed"}
17
+ {"time":"2026-07-21T17:56:08.143167424Z","level":"INFO","msg":"sender: closed"}
18
+ {"time":"2026-07-21T17:56:08.143170884Z","level":"INFO","msg":"stream: all finished"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/config.yaml ADDED
@@ -0,0 +1,332 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.27.2
4
+ e:
5
+ jh73o7nak8ocbfkoinc841i4xrfesdiu:
6
+ args:
7
+ - --config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json
8
+ - --resume=true
9
+ - --dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
10
+ - --output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
11
+ codePath: benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
12
+ codePathLocal: my_scripts/lerobot_train_yaml.py
13
+ cpu_count: 112
14
+ cpu_count_logical: 224
15
+ cudaVersion: "12.8"
16
+ disk:
17
+ /:
18
+ total: "1888425144320"
19
+ used: "1061565542400"
20
+ executable: /home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python
21
+ git:
22
+ commit: c1ff610ae4b02827f1c984cef8bbf28a7889bd6f
23
+ remote: git@github.com:paragon7060/clvla.git
24
+ gpu: NVIDIA H200
25
+ gpu_count: 2
26
+ gpu_nvidia:
27
+ - architecture: Hopper
28
+ cudaCores: 16896
29
+ memoryTotal: "150754820096"
30
+ name: NVIDIA H200
31
+ uuid: GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991
32
+ - architecture: Hopper
33
+ cudaCores: 16896
34
+ memoryTotal: "150754820096"
35
+ name: NVIDIA H200
36
+ uuid: GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc
37
+ host: ext_minje-box
38
+ memory:
39
+ total: "2164176568320"
40
+ os: Linux-6.8.0-55-generic-x86_64-with-glibc2.39
41
+ program: /home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
42
+ python: CPython 3.12.13
43
+ root: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
44
+ startedAt: "2026-07-21T18:02:57.910678Z"
45
+ writerId: jh73o7nak8ocbfkoinc841i4xrfesdiu
46
+ m: []
47
+ python_version: 3.12.13
48
+ t:
49
+ "1":
50
+ - 1
51
+ - 11
52
+ - 41
53
+ - 49
54
+ - 51
55
+ - 71
56
+ - 83
57
+ - 98
58
+ "2":
59
+ - 1
60
+ - 11
61
+ - 41
62
+ - 49
63
+ - 51
64
+ - 71
65
+ - 83
66
+ - 98
67
+ "3":
68
+ - 5
69
+ - 13
70
+ - 14
71
+ - 15
72
+ - 16
73
+ - 62
74
+ "4": 3.12.13
75
+ "5": 0.27.2
76
+ "6": 5.5.4
77
+ "10":
78
+ - 21
79
+ "12": 0.27.2
80
+ "13": linux-x86_64
81
+ batch_size:
82
+ value: 128
83
+ checkpoint_path:
84
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000
85
+ cudnn_deterministic:
86
+ value: false
87
+ dataset:
88
+ value:
89
+ depth_output_unit: mm
90
+ episodes: null
91
+ eval_split: 0
92
+ image_transforms:
93
+ enable: false
94
+ max_num_transforms: 3
95
+ random_order: false
96
+ tfs:
97
+ affine:
98
+ kwargs:
99
+ degrees:
100
+ - -5
101
+ - 5
102
+ translate:
103
+ - 0.05
104
+ - 0.05
105
+ type: RandomAffine
106
+ weight: 1
107
+ brightness:
108
+ kwargs:
109
+ brightness:
110
+ - 0.8
111
+ - 1.2
112
+ type: ColorJitter
113
+ weight: 1
114
+ contrast:
115
+ kwargs:
116
+ contrast:
117
+ - 0.8
118
+ - 1.2
119
+ type: ColorJitter
120
+ weight: 1
121
+ hue:
122
+ kwargs:
123
+ hue:
124
+ - -0.05
125
+ - 0.05
126
+ type: ColorJitter
127
+ weight: 1
128
+ saturation:
129
+ kwargs:
130
+ saturation:
131
+ - 0.5
132
+ - 1.5
133
+ type: ColorJitter
134
+ weight: 1
135
+ sharpness:
136
+ kwargs:
137
+ sharpness:
138
+ - 0.5
139
+ - 1.5
140
+ type: SharpnessJitter
141
+ weight: 1
142
+ repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
143
+ return_uint8: false
144
+ revision: null
145
+ root: /home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
146
+ streaming: false
147
+ use_imagenet_stats: true
148
+ video_backend: torchcodec
149
+ env:
150
+ value: null
151
+ env_eval_freq:
152
+ value: 20000
153
+ eval:
154
+ value:
155
+ batch_size: 50
156
+ n_episodes: 50
157
+ recording: false
158
+ recording_private: false
159
+ recording_repo_id: null
160
+ use_async_envs: true
161
+ eval_steps:
162
+ value: 0
163
+ job:
164
+ value:
165
+ detach: false
166
+ image: huggingface/lerobot-gpu:latest
167
+ tags: []
168
+ target: null
169
+ timeout: 2d
170
+ job_name:
171
+ value: INSIGHT_RD_RKD_TimewarpVAE_RSCLstyle_C60
172
+ log_freq:
173
+ value: 200
174
+ max_eval_samples:
175
+ value: 0
176
+ num_workers:
177
+ value: 4
178
+ optimizer:
179
+ value:
180
+ betas:
181
+ - 0.9
182
+ - 0.999
183
+ eps: 1e-08
184
+ grad_clip_norm: 1
185
+ lr: 0.0001
186
+ type: adamw
187
+ weight_decay: 1e-05
188
+ output_dir:
189
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
190
+ peft:
191
+ value: null
192
+ persistent_workers:
193
+ value: true
194
+ policy:
195
+ value:
196
+ action_decode_transform: null
197
+ balance_dataset_weights: true
198
+ balance_trajectory_weights: true
199
+ base_model_path: nvidia/GR00T-N1.7-3B
200
+ batch_size: 32
201
+ chunk_size: 16
202
+ dataloader_num_workers: 8
203
+ dataset_paths: null
204
+ device: cuda
205
+ embodiment_tag: new_embodiment
206
+ image_size:
207
+ - 256
208
+ - 256
209
+ input_features:
210
+ observation.images.guide:
211
+ shape:
212
+ - 3
213
+ - 224
214
+ - 224
215
+ type: VISUAL
216
+ observation.images.right_shoulder:
217
+ shape:
218
+ - 3
219
+ - 224
220
+ - 224
221
+ type: VISUAL
222
+ observation.images.wrist:
223
+ shape:
224
+ - 3
225
+ - 224
226
+ - 224
227
+ type: VISUAL
228
+ observation.state:
229
+ shape:
230
+ - 16
231
+ type: STATE
232
+ license: null
233
+ lora_alpha: 16
234
+ lora_dropout: 0.1
235
+ lora_full_model: false
236
+ lora_rank: 0
237
+ max_action_dim: 132
238
+ max_state_dim: 132
239
+ max_steps: 10000
240
+ model_params_fp32: true
241
+ n_action_steps: 16
242
+ n_obs_steps: 1
243
+ normalization_mapping:
244
+ ACTION: IDENTITY
245
+ STATE: IDENTITY
246
+ VISUAL: IDENTITY
247
+ num_inference_timesteps: null
248
+ optimizer_betas:
249
+ - 0.9
250
+ - 0.999
251
+ optimizer_eps: 1e-08
252
+ optimizer_lr: 0.0001
253
+ optimizer_weight_decay: 1e-05
254
+ output_dir: ./tmp/gr00t
255
+ output_features:
256
+ action:
257
+ shape:
258
+ - 16
259
+ type: ACTION
260
+ pretrained_path: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
261
+ pretrained_revision: null
262
+ private: null
263
+ push_to_hub: false
264
+ relative_exclude_joints: []
265
+ repo_id: null
266
+ report_to: wandb
267
+ resume: false
268
+ rkd_angle_weight: 2
269
+ rkd_distance_weight: 1
270
+ rkd_enabled: true
271
+ rkd_fm_loss_weight: 1
272
+ rkd_loss_weight: 1
273
+ rkd_student_attention_hidden_dim: 512
274
+ rkd_student_attention_output_dim: 512
275
+ rkd_student_pooling: attention
276
+ rkd_vae_checkpoint: /home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84
277
+ rtc_ramp_rate: null
278
+ save_steps: 1000
279
+ tags: null
280
+ tokenizer_assets_repo: null
281
+ tune_diffusion_model: true
282
+ tune_llm: false
283
+ tune_projector: true
284
+ tune_top_llm_layers: 0
285
+ tune_visual: false
286
+ tune_vlln: true
287
+ type: groot_rkd
288
+ use_amp: false
289
+ use_bf16: true
290
+ use_flash_attention: false
291
+ use_peft: false
292
+ use_relative_actions: false
293
+ video_backend: decord
294
+ warmup_ratio: 0.05
295
+ warmup_steps: 3000
296
+ prefetch_factor:
297
+ value: 4
298
+ resume:
299
+ value: true
300
+ reward_model:
301
+ value: null
302
+ sample_weighting:
303
+ value: null
304
+ save_checkpoint:
305
+ value: true
306
+ save_checkpoint_to_hub:
307
+ value: false
308
+ save_freq:
309
+ value: 20000
310
+ scheduler:
311
+ value:
312
+ name: cosine
313
+ num_warmup_steps: 3000
314
+ type: diffuser
315
+ seed:
316
+ value: 42
317
+ steps:
318
+ value: 60000
319
+ tolerance_s:
320
+ value: 0.0001
321
+ use_policy_training_preset:
322
+ value: true
323
+ wandb:
324
+ value:
325
+ add_tags: true
326
+ disable_artifact: true
327
+ enable: true
328
+ entity: null
329
+ mode: null
330
+ notes: null
331
+ project: lerobot
332
+ run_id: bepgb8uu
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/wandb-metadata.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-55-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-07-21T18:02:57.910678Z",
5
+ "args": [
6
+ "--config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json",
7
+ "--resume=true",
8
+ "--dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2",
9
+ "--output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K"
10
+ ],
11
+ "program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
12
+ "codePath": "benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
13
+ "codePathLocal": "my_scripts/lerobot_train_yaml.py",
14
+ "git": {
15
+ "remote": "git@github.com:paragon7060/clvla.git",
16
+ "commit": "c1ff610ae4b02827f1c984cef8bbf28a7889bd6f"
17
+ },
18
+ "root": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K",
19
+ "host": "ext_minje-box",
20
+ "executable": "/home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python",
21
+ "cpu_count": 112,
22
+ "cpu_count_logical": 224,
23
+ "gpu": "NVIDIA H200",
24
+ "gpu_count": 2,
25
+ "disk": {
26
+ "/": {
27
+ "total": "1888425144320",
28
+ "used": "1061565542400"
29
+ }
30
+ },
31
+ "memory": {
32
+ "total": "2164176568320"
33
+ },
34
+ "gpu_nvidia": [
35
+ {
36
+ "name": "NVIDIA H200",
37
+ "memoryTotal": "150754820096",
38
+ "cudaCores": 16896,
39
+ "architecture": "Hopper",
40
+ "uuid": "GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991"
41
+ },
42
+ {
43
+ "name": "NVIDIA H200",
44
+ "memoryTotal": "150754820096",
45
+ "cudaCores": 16896,
46
+ "architecture": "Hopper",
47
+ "uuid": "GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc"
48
+ }
49
+ ],
50
+ "cudaVersion": "12.8",
51
+ "writerId": "jh73o7nak8ocbfkoinc841i4xrfesdiu"
52
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/logs/debug-core.log ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T18:02:57.941059216Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpymhukzhl/port-2613.txt","pid":2613,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-07-21T18:02:57.941546044Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-2613-3208-3437201184/socket","Net":"unix"}}
3
+ {"time":"2026-07-21T18:02:57.941599071Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":2613}
4
+ {"time":"2026-07-21T18:02:57.960451132Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpw2e5f7tu/port-2603.txt","pid":2603,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
5
+ {"time":"2026-07-21T18:02:57.961187724Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":2603}
6
+ {"time":"2026-07-21T18:02:57.961187959Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-2603-3232-1020014156/socket","Net":"unix"}}
7
+ {"time":"2026-07-21T18:02:57.997202494Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpzsgp20sc/port-2608.txt","pid":2608,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
8
+ {"time":"2026-07-21T18:02:57.997759959Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":2608}
9
+ {"time":"2026-07-21T18:02:57.997700668Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-2608-3257-2579777137/socket","Net":"unix"}}
10
+ {"time":"2026-07-21T18:02:58.127169649Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
11
+ {"time":"2026-07-21T18:02:58.133811039Z","level":"INFO","msg":"handleInformInit: received","streamId":"bepgb8uu","id":"1(@)"}
12
+ {"time":"2026-07-21T18:02:58.147363472Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
13
+ {"time":"2026-07-21T18:02:58.154671515Z","level":"INFO","msg":"handleInformInit: received","streamId":"r3cz38bm","id":"1(@)"}
14
+ {"time":"2026-07-21T18:02:58.183360713Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
15
+ {"time":"2026-07-21T18:02:58.190338084Z","level":"INFO","msg":"handleInformInit: received","streamId":"ny0tx2de","id":"1(@)"}
16
+ {"time":"2026-07-21T18:02:58.595769834Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"bepgb8uu","id":"1(@)"}
17
+ {"time":"2026-07-21T18:02:58.603119658Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"r3cz38bm","id":"1(@)"}
18
+ {"time":"2026-07-21T18:02:58.63670199Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"ny0tx2de","id":"1(@)"}
19
+ {"time":"2026-07-21T18:03:04.30513956Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"yck62n1u63yx"}
20
+ {"time":"2026-07-21T18:03:04.306532835Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"7w4l5mjhl37y"}
21
+ {"time":"2026-07-21T18:03:04.608540167Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"6qdbap31ko3x"}
22
+ {"time":"2026-07-21T18:03:20.637644594Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
23
+ {"time":"2026-07-21T18:03:20.637700863Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
24
+ {"time":"2026-07-21T18:03:20.637726007Z","level":"INFO","msg":"server is shutting down"}
25
+ {"time":"2026-07-21T18:03:20.637766195Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
26
+ {"time":"2026-07-21T18:03:20.637774602Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
27
+ {"time":"2026-07-21T18:03:20.637845762Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-2603-3232-1020014156/socket","Net":"unix"}}
28
+ {"time":"2026-07-21T18:03:21.443933528Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
29
+ {"time":"2026-07-21T18:03:21.444100871Z","level":"INFO","msg":"server is shutting down"}
30
+ {"time":"2026-07-21T18:03:21.444201551Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-2608-3257-2579777137/socket","Net":"unix"}}
31
+ {"time":"2026-07-21T18:03:21.444210719Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
32
+ {"time":"2026-07-21T18:03:21.444257407Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
33
+ {"time":"2026-07-21T18:03:21.444322814Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
34
+ {"time":"2026-07-21T18:03:21.847622296Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
35
+ {"time":"2026-07-21T18:03:21.847643396Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
36
+ {"time":"2026-07-21T18:03:21.847655052Z","level":"INFO","msg":"server is closed"}
37
+ {"time":"2026-07-21T18:03:22.171687103Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
38
+ {"time":"2026-07-21T18:03:22.17174701Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
39
+ {"time":"2026-07-21T18:03:22.171781188Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
40
+ {"time":"2026-07-21T18:03:22.171810691Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
41
+ {"time":"2026-07-21T18:03:22.17176302Z","level":"INFO","msg":"server is shutting down"}
42
+ {"time":"2026-07-21T18:03:22.17190467Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-2613-3208-3437201184/socket","Net":"unix"}}
43
+ {"time":"2026-07-21T18:03:23.465220754Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
44
+ {"time":"2026-07-21T18:03:23.465248434Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
45
+ {"time":"2026-07-21T18:03:23.465257209Z","level":"INFO","msg":"server is closed"}
46
+ {"time":"2026-07-21T18:03:35.421783314Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
47
+ {"time":"2026-07-21T18:03:35.421805809Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
48
+ {"time":"2026-07-21T18:03:35.421814159Z","level":"INFO","msg":"server is closed"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/config.yaml ADDED
@@ -0,0 +1,332 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.27.2
4
+ e:
5
+ n5i65e3ba3p72ehycvvdbj4pqt8mvnt8:
6
+ args:
7
+ - --config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json
8
+ - --resume=true
9
+ - --dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
10
+ - --output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
11
+ codePath: benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
12
+ codePathLocal: my_scripts/lerobot_train_yaml.py
13
+ cpu_count: 112
14
+ cpu_count_logical: 224
15
+ cudaVersion: "12.8"
16
+ disk:
17
+ /:
18
+ total: "1888425144320"
19
+ used: "1061567148032"
20
+ executable: /home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python
21
+ git:
22
+ commit: c1ff610ae4b02827f1c984cef8bbf28a7889bd6f
23
+ remote: git@github.com:paragon7060/clvla.git
24
+ gpu: NVIDIA H200
25
+ gpu_count: 2
26
+ gpu_nvidia:
27
+ - architecture: Hopper
28
+ cudaCores: 16896
29
+ memoryTotal: "150754820096"
30
+ name: NVIDIA H200
31
+ uuid: GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991
32
+ - architecture: Hopper
33
+ cudaCores: 16896
34
+ memoryTotal: "150754820096"
35
+ name: NVIDIA H200
36
+ uuid: GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc
37
+ host: ext_minje-box
38
+ memory:
39
+ total: "2164176568320"
40
+ os: Linux-6.8.0-55-generic-x86_64-with-glibc2.39
41
+ program: /home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py
42
+ python: CPython 3.12.13
43
+ root: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
44
+ startedAt: "2026-07-21T18:07:58.689632Z"
45
+ writerId: n5i65e3ba3p72ehycvvdbj4pqt8mvnt8
46
+ m: []
47
+ python_version: 3.12.13
48
+ t:
49
+ "1":
50
+ - 1
51
+ - 11
52
+ - 41
53
+ - 49
54
+ - 51
55
+ - 71
56
+ - 83
57
+ - 98
58
+ "2":
59
+ - 1
60
+ - 11
61
+ - 41
62
+ - 49
63
+ - 51
64
+ - 71
65
+ - 83
66
+ - 98
67
+ "3":
68
+ - 5
69
+ - 13
70
+ - 14
71
+ - 15
72
+ - 16
73
+ - 62
74
+ "4": 3.12.13
75
+ "5": 0.27.2
76
+ "6": 5.5.4
77
+ "10":
78
+ - 21
79
+ "12": 0.27.2
80
+ "13": linux-x86_64
81
+ batch_size:
82
+ value: 128
83
+ checkpoint_path:
84
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000
85
+ cudnn_deterministic:
86
+ value: false
87
+ dataset:
88
+ value:
89
+ depth_output_unit: mm
90
+ episodes: null
91
+ eval_split: 0
92
+ image_transforms:
93
+ enable: false
94
+ max_num_transforms: 3
95
+ random_order: false
96
+ tfs:
97
+ affine:
98
+ kwargs:
99
+ degrees:
100
+ - -5
101
+ - 5
102
+ translate:
103
+ - 0.05
104
+ - 0.05
105
+ type: RandomAffine
106
+ weight: 1
107
+ brightness:
108
+ kwargs:
109
+ brightness:
110
+ - 0.8
111
+ - 1.2
112
+ type: ColorJitter
113
+ weight: 1
114
+ contrast:
115
+ kwargs:
116
+ contrast:
117
+ - 0.8
118
+ - 1.2
119
+ type: ColorJitter
120
+ weight: 1
121
+ hue:
122
+ kwargs:
123
+ hue:
124
+ - -0.05
125
+ - 0.05
126
+ type: ColorJitter
127
+ weight: 1
128
+ saturation:
129
+ kwargs:
130
+ saturation:
131
+ - 0.5
132
+ - 1.5
133
+ type: ColorJitter
134
+ weight: 1
135
+ sharpness:
136
+ kwargs:
137
+ sharpness:
138
+ - 0.5
139
+ - 1.5
140
+ type: SharpnessJitter
141
+ weight: 1
142
+ repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
143
+ return_uint8: false
144
+ revision: null
145
+ root: /home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
146
+ streaming: false
147
+ use_imagenet_stats: true
148
+ video_backend: torchcodec
149
+ env:
150
+ value: null
151
+ env_eval_freq:
152
+ value: 20000
153
+ eval:
154
+ value:
155
+ batch_size: 50
156
+ n_episodes: 50
157
+ recording: false
158
+ recording_private: false
159
+ recording_repo_id: null
160
+ use_async_envs: true
161
+ eval_steps:
162
+ value: 0
163
+ job:
164
+ value:
165
+ detach: false
166
+ image: huggingface/lerobot-gpu:latest
167
+ tags: []
168
+ target: null
169
+ timeout: 2d
170
+ job_name:
171
+ value: INSIGHT_RD_RKD_TimewarpVAE_RSCLstyle_C60
172
+ log_freq:
173
+ value: 200
174
+ max_eval_samples:
175
+ value: 0
176
+ num_workers:
177
+ value: 4
178
+ optimizer:
179
+ value:
180
+ betas:
181
+ - 0.9
182
+ - 0.999
183
+ eps: 1e-08
184
+ grad_clip_norm: 1
185
+ lr: 0.0001
186
+ type: adamw
187
+ weight_decay: 1e-05
188
+ output_dir:
189
+ value: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
190
+ peft:
191
+ value: null
192
+ persistent_workers:
193
+ value: true
194
+ policy:
195
+ value:
196
+ action_decode_transform: null
197
+ balance_dataset_weights: true
198
+ balance_trajectory_weights: true
199
+ base_model_path: nvidia/GR00T-N1.7-3B
200
+ batch_size: 32
201
+ chunk_size: 16
202
+ dataloader_num_workers: 8
203
+ dataset_paths: null
204
+ device: cuda
205
+ embodiment_tag: new_embodiment
206
+ image_size:
207
+ - 256
208
+ - 256
209
+ input_features:
210
+ observation.images.guide:
211
+ shape:
212
+ - 3
213
+ - 224
214
+ - 224
215
+ type: VISUAL
216
+ observation.images.right_shoulder:
217
+ shape:
218
+ - 3
219
+ - 224
220
+ - 224
221
+ type: VISUAL
222
+ observation.images.wrist:
223
+ shape:
224
+ - 3
225
+ - 224
226
+ - 224
227
+ type: VISUAL
228
+ observation.state:
229
+ shape:
230
+ - 16
231
+ type: STATE
232
+ license: null
233
+ lora_alpha: 16
234
+ lora_dropout: 0.1
235
+ lora_full_model: false
236
+ lora_rank: 0
237
+ max_action_dim: 132
238
+ max_state_dim: 132
239
+ max_steps: 10000
240
+ model_params_fp32: true
241
+ n_action_steps: 16
242
+ n_obs_steps: 1
243
+ normalization_mapping:
244
+ ACTION: IDENTITY
245
+ STATE: IDENTITY
246
+ VISUAL: IDENTITY
247
+ num_inference_timesteps: null
248
+ optimizer_betas:
249
+ - 0.9
250
+ - 0.999
251
+ optimizer_eps: 1e-08
252
+ optimizer_lr: 0.0001
253
+ optimizer_weight_decay: 1e-05
254
+ output_dir: ./tmp/gr00t
255
+ output_features:
256
+ action:
257
+ shape:
258
+ - 16
259
+ type: ACTION
260
+ pretrained_path: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
261
+ pretrained_revision: null
262
+ private: null
263
+ push_to_hub: false
264
+ relative_exclude_joints: []
265
+ repo_id: null
266
+ report_to: wandb
267
+ resume: false
268
+ rkd_angle_weight: 2
269
+ rkd_distance_weight: 1
270
+ rkd_enabled: true
271
+ rkd_fm_loss_weight: 1
272
+ rkd_loss_weight: 1
273
+ rkd_student_attention_hidden_dim: 512
274
+ rkd_student_attention_output_dim: 512
275
+ rkd_student_pooling: attention
276
+ rkd_vae_checkpoint: /home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84
277
+ rtc_ramp_rate: null
278
+ save_steps: 1000
279
+ tags: null
280
+ tokenizer_assets_repo: null
281
+ tune_diffusion_model: true
282
+ tune_llm: false
283
+ tune_projector: true
284
+ tune_top_llm_layers: 0
285
+ tune_visual: false
286
+ tune_vlln: true
287
+ type: groot_rkd
288
+ use_amp: false
289
+ use_bf16: true
290
+ use_flash_attention: false
291
+ use_peft: false
292
+ use_relative_actions: false
293
+ video_backend: decord
294
+ warmup_ratio: 0.05
295
+ warmup_steps: 3000
296
+ prefetch_factor:
297
+ value: 4
298
+ resume:
299
+ value: true
300
+ reward_model:
301
+ value: null
302
+ sample_weighting:
303
+ value: null
304
+ save_checkpoint:
305
+ value: true
306
+ save_checkpoint_to_hub:
307
+ value: false
308
+ save_freq:
309
+ value: 20000
310
+ scheduler:
311
+ value:
312
+ name: cosine
313
+ num_warmup_steps: 3000
314
+ type: diffuser
315
+ seed:
316
+ value: 42
317
+ steps:
318
+ value: 60000
319
+ tolerance_s:
320
+ value: 0.0001
321
+ use_policy_training_preset:
322
+ value: true
323
+ wandb:
324
+ value:
325
+ add_tags: true
326
+ disable_artifact: true
327
+ enable: true
328
+ entity: null
329
+ mode: null
330
+ notes: null
331
+ project: lerobot
332
+ run_id: bepgb8uu
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/output.log ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2026-07-21 18:07:59 db_utils.py:121 Logs will be synced with wandb.
2
+ INFO 2026-07-21 18:07:59 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/bepgb8uu
3
+ INFO 2026-07-21 18:07:59 ot_train.py:298 Creating dataset
4
+ INFO 2026-07-21 18:08:01 ot_train.py:332 Creating policy
5
+ INFO 2026-07-21 18:08:01 ng_groot.py:191 The Groot policy wraps NVIDIA's GR00T n1.7 model. Loading pretrained model from: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
6
+ INFO 2026-07-21 18:08:01 ng_groot.py:225 Detected fine-tuned LeRobot checkpoint, loading with state dict...
7
+ Fetching 27 files: 100%|██████████| 27/27 [00:00<00:00, 4674.19it/s]
8
+ `torch_dtype` is deprecated! Use `dtype` instead!
9
+ Loading weights: 100%|██████████| 1031/1031 [00:00<00:00, 1234.07it/s]
10
+ Loading weights from local directory
11
+ Traceback (most recent call last):
12
+ File "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py", line 1430, in <module>
13
+ main()
14
+ File "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py", line 1426, in main
15
+ train_module.main()
16
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/scripts/lerobot_train.py", line 816, in main
17
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/configs/parser.py", line 320, in wrapper_inner
18
+ response = fn(cfg, *args, **kwargs)
19
+ ^^^^^^^^^^^^^^^^^^^^^^^^
20
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/scripts/lerobot_train.py", line 397, in train
21
+ logging.info(colored("Output dir:", "yellow", attrs=["bold"]) + f" {cfg.output_dir}")
22
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
23
+ File "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py", line 1204, in make_pre_post_processors
24
+ return original_make_pre_post_processors(*args, **kwargs)
25
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
26
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/policies/factory.py", line 323, in make_pre_post_processors
27
+ preprocessor = PolicyProcessorPipeline.from_pretrained(
28
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
29
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/processor/pipeline.py", line 738, in from_pretrained
30
+ cls._validate_overrides_used(validated_overrides, loaded_config)
31
+ File "/home/ext_minje/clvla/lerobot/src/lerobot/processor/pipeline.py", line 1254, in _validate_overrides_used
32
+ raise KeyError(
33
+ KeyError: "Override keys ['normalizer_processor'] do not match any step in the saved configuration. Available step keys: ['rename_observations_processor', 'to_batch_processor', 'rkd_preserve_raw_action', 'groot_n1_7_pack_inputs_v1', 'groot_n1_7_vlm_encode_v1', 'device_processor']. Make sure override keys match exact step class names or registry names."
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"train/epochs":23.212298561595844,"train/samples":5068800,"train/grad_norm":0.11059212811291218,"train/rkd_fm_loss_weight":1,"train/training_phase_index":0,"train/samples_per_s":40.698668116487305,"train/rkd_student_feature_norm":11.796961784362791,"train/gpu_mem_gb":52.89515282392502,"_runtime":120057,"train/dataloading_s":1.5435121013406023,"train/training_phase_until_step":60000,"train/rkd_loss":0.0015607024542987349,"train/training_phase_start_step":0,"train/loss":0.0115696769207716,"train/rkd_loss_weight":0.2591461049345566,"_timestamp":1.7845906275011618e+09,"train/rkd_angle_loss":0.0006215126486495137,"train/rkd_distance_loss":0.0003176771861035377,"train/rkd_teacher_latent_norm":1,"train/rkd_runtime_enabled":1,"train/steps":39600,"train/weighted_rkd_loss":0.0004044499364681542,"_step":39600,"train/rkd_student_pooling_id":1,"train/episodes":114436.63190866752,"train/training_phase_trainable_params":1.622090881e+09,"train/lr":2.8660040166848404e-05,"train/weighted_flow_matching_loss":0.011165226809680462,"train/update_s":1.601553964055929,"_wandb":{"runtime":120057},"train/flow_matching_loss":0.011165226809680462}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/logs/debug.log ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
2
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_setup.py:_flush():81] Configure stats pid to 5579
3
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/logs/debug.log
5
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/logs/debug-internal.log
6
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_init.py:init():766] calling init triggers
7
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_init.py:init():771] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'Whalswp/INSIGHTfixposV4_filtered_multispace_v2', 'root': '/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'return_uint8': False, 'depth_output_unit': 'mm', 'streaming': False, 'eval_split': 0.0}, 'env': None, 'policy': {'type': 'groot_rkd', 'n_obs_steps': 1, 'input_features': {'observation.state': {'type': <FeatureType.STATE: 'STATE'>, 'shape': [16]}, 'observation.images.wrist': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}, 'observation.images.right_shoulder': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}, 'observation.images.guide': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}}, 'output_features': {'action': {'type': <FeatureType.ACTION: 'ACTION'>, 'shape': [16]}}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model', 'pretrained_revision': None, 'chunk_size': 16, 'n_action_steps': 16, 'max_state_dim': 132, 'max_action_dim': 132, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'ACTION': <NormalizationMode.IDENTITY: 'IDENTITY'>}, 'base_model_path': 'nvidia/GR00T-N1.7-3B', 'action_decode_transform': None, 'embodiment_tag': 'new_embodiment', 'tune_llm': False, 'tune_visual': False, 'tune_projector': True, 'tune_diffusion_model': True, 'tune_vlln': True, 'tune_top_llm_layers': 0, 'num_inference_timesteps': None, 'rtc_ramp_rate': None, 'use_flash_attention': False, 'use_relative_actions': False, 'relative_exclude_joints': [], 'optimizer_lr': 0.0001, 'optimizer_betas': [0.9, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-05, 'warmup_steps': 3000, 'warmup_ratio': 0.05, 'use_bf16': True, 'model_params_fp32': True, 'image_size': [256, 256], 'tokenizer_assets_repo': None, 'lora_rank': 0, 'lora_alpha': 16, 'lora_dropout': 0.1, 'lora_full_model': False, 'video_backend': 'decord', 'balance_dataset_weights': True, 'balance_trajectory_weights': True, 'dataset_paths': None, 'output_dir': './tmp/gr00t', 'save_steps': 1000, 'max_steps': 10000, 'batch_size': 32, 'dataloader_num_workers': 8, 'report_to': 'wandb', 'resume': False, 'rkd_enabled': True, 'rkd_vae_checkpoint': '/home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84', 'rkd_student_pooling': 'attention', 'rkd_student_attention_hidden_dim': 512, 'rkd_student_attention_output_dim': 512, 'rkd_fm_loss_weight': 1.0, 'rkd_loss_weight': 1.0, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 2.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K', 'job_name': 'INSIGHT_RD_RKD_TimewarpVAE_RSCLstyle_C60', 'resume': True, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 128, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 60000, 'env_eval_freq': 20000, 'log_freq': 200, 'eval_steps': 0, 'max_eval_samples': 0, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 0.0001, 'weight_decay': 1e-05, 'grad_clip_norm': 1.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 3000, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': True, 'recording': False, 'recording_repo_id': None, 'recording_private': False}, 'wandb': {'enable': True, 'disable_artifact': True, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': 'bepgb8uu', 'mode': None, 'add_tags': True}, 'peft': None, 'job': {'target': None, 'image': 'huggingface/lerobot-gpu:latest', 'timeout': '2d', 'detach': False, 'tags': []}, 'save_checkpoint_to_hub': False, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000', '_wandb': {}}
9
+ 2026-07-21 18:07:58,690 INFO MainThread:5579 [wandb_init.py:init():814] starting backend
10
+ 2026-07-21 18:07:58,906 INFO MainThread:5579 [wandb_init.py:init():829] sending inform_init request
11
+ 2026-07-21 18:07:59,270 INFO MainThread:5579 [wandb_init.py:init():834] backend started and connected
12
+ 2026-07-21 18:07:59,272 INFO MainThread:5579 [wandb_init.py:init():904] updated telemetry
13
+ 2026-07-21 18:07:59,276 INFO MainThread:5579 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
14
+ 2026-07-21 18:07:59,859 INFO MainThread:5579 [wandb_init.py:init():967] run resumed
15
+ 2026-07-21 18:07:59,861 INFO MainThread:5579 [wandb_init.py:init():972] starting run threads in backend
16
+ 2026-07-21 18:07:59,912 INFO MainThread:5579 [wandb_run.py:_console_start():2523] atexit reg
17
+ 2026-07-21 18:07:59,912 INFO MainThread:5579 [wandb_run.py:_redirect():2373] redirect: wrap_raw
18
+ 2026-07-21 18:07:59,912 INFO MainThread:5579 [wandb_run.py:_redirect():2442] Wrapping output streams.
19
+ 2026-07-21 18:07:59,912 INFO MainThread:5579 [wandb_run.py:_redirect():2465] Redirects installed.
20
+ 2026-07-21 18:07:59,913 INFO MainThread:5579 [wandb_init.py:init():1010] run started, returning control to user process
21
+ 2026-07-21 18:08:20,850 INFO wandb-AsyncioManager-main:5579 [service_client.py:_forward_responses():122] Reached EOF.
22
+ 2026-07-21 18:08:20,851 INFO wandb-AsyncioManager-main:5579 [mailbox.py:close():154] Closing mailbox, abandoning 2 handles.
23
+ 2026-07-21 18:08:20,851 ERROR wandb-AsyncioManager-main:5579 [asyncio_manager.py:fn_wrap_exceptions():184] Uncaught exception in run_soon callback.
24
+ Traceback (most recent call last):
25
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/asyncio_manager.py", line 182, in fn_wrap_exceptions
26
+ await fn()
27
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/run_messages.py", line 90, in loop
28
+ await asyncio_compat.race(
29
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/asyncio_compat.py", line 278, in race
30
+ async with open_task_group(race=True) as tg:
31
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^
32
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/contextlib.py", line 217, in __aexit__
33
+ await anext(self.gen)
34
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/asyncio_compat.py", line 240, in open_task_group
35
+ await task_group._wait_all(race=race, timeout=exit_timeout)
36
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/asyncio_compat.py", line 180, in _wait_all
37
+ raise exc
38
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/lib/run_messages.py", line 113, in _print_all
39
+ result = await handle.wait_async(timeout=None)
40
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
41
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/mailbox/mailbox_handle.py", line 127, in wait_async
42
+ response = await self._handle.wait_async(timeout=timeout)
43
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
44
+ File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/wandb/sdk/mailbox/response_handle.py", line 123, in wait_async
45
+ raise HandleAbandonedError()
46
+ wandb.sdk.mailbox.mailbox_handle.HandleAbandonedError
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/output.log ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2026-07-21 18:14:25 db_utils.py:121 Logs will be synced with wandb.
2
+ INFO 2026-07-21 18:14:25 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/bepgb8uu
3
+ INFO 2026-07-21 18:14:25 ot_train.py:298 Creating dataset
4
+ INFO 2026-07-21 18:14:26 ot_train.py:332 Creating policy
5
+ INFO 2026-07-21 18:14:26 ng_groot.py:191 The Groot policy wraps NVIDIA's GR00T n1.7 model. Loading pretrained model from: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
6
+ INFO 2026-07-21 18:14:26 ng_groot.py:225 Detected fine-tuned LeRobot checkpoint, loading with state dict...
7
+ Fetching 27 files: 100%|██████████| 27/27 [00:00<00:00, 3045.07it/s]
8
+ `torch_dtype` is deprecated! Use `dtype` instead!
9
+ Loading weights: 100%|██████████| 1031/1031 [00:00<00:00, 1088.13it/s]
10
+ Loading weights from local directory
11
+ INFO 2026-07-21 18:14:38 ot_train.py:405 Creating optimizer and scheduler
12
+ INFO 2026-07-21 18:14:44 ot_train.py:439 Output dir: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
13
+ INFO 2026-07-21 18:14:44 ot_train.py:446 cfg.steps=60000 (60K)
14
+ INFO 2026-07-21 18:14:44 ot_train.py:447 dataset.num_frames=218367 (218K)
15
+ INFO 2026-07-21 18:14:44 ot_train.py:448 dataset.num_episodes=4930
16
+ INFO 2026-07-21 18:14:44 ot_train.py:451 Effective batch size: 128 x 1 = 128
17
+ INFO 2026-07-21 18:14:44 ot_train.py:452 num_learnable_params=1622090881 (2B)
18
+ INFO 2026-07-21 18:14:44 ot_train.py:453 num_total_params=3145712033 (3B)
19
+ INFO 2026-07-21 18:14:44 ot_train.py:494 Resuming data order at epoch 26, sample 82688
20
+ Training: 0%| | 0/30000 [00:00<?, ?step/s]INFO 2026-07-21 18:14:44 ot_train.py:604 Start offline training on a fixed dataset, with effective batch size: 128
21
+ Training: 0%| | 109/30000 [04:49<22:21:57, 2.69s/step]
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/requirements.txt ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ tzdata==2026.3
2
+ nvidia-cufile-cu12==1.13.1.3
3
+ wandb==0.27.2
4
+ six==1.17.0
5
+ peft==0.19.1
6
+ nvidia-curand-cu12==10.3.9.90
7
+ nvidia-cuda-runtime-cu12==12.8.90
8
+ datasets==4.8.5
9
+ attrs==26.1.0
10
+ cloudpickle==3.1.2
11
+ packaging==25.0
12
+ nvidia-cublas-cu12==12.8.4.1
13
+ cmake==4.1.3
14
+ psutil==7.2.2
15
+ pydantic==2.13.4
16
+ GitPython==3.1.51
17
+ pillow==12.2.0
18
+ pytz==2026.2
19
+ cuda-pathfinder==1.2.2
20
+ nvidia-cusparse-cu12==12.5.8.93
21
+ typing_extensions==4.15.0
22
+ urllib3==2.7.0
23
+ pydantic_core==2.46.4
24
+ draccus==0.10.0
25
+ mypy_extensions==1.1.0
26
+ dill==0.4.1
27
+ typer==0.26.8
28
+ decord==0.6.0
29
+ tqdm==4.68.4
30
+ triton==3.6.0
31
+ idna==3.18
32
+ requests==2.34.2
33
+ nvidia-cudnn-cu12==9.19.0.56
34
+ mdurl==0.1.2
35
+ shellingham==1.5.4
36
+ zipp==4.1.0
37
+ timm==1.0.28
38
+ torch==2.11.0+cu128
39
+ tokenizers==0.22.2
40
+ nvidia-cusparselt-cu12==0.7.1
41
+ nvidia-cufft-cu12==11.3.3.83
42
+ nvidia-cusolver-cu12==11.7.3.90
43
+ opencv-python-headless==4.13.0.92
44
+ httpx==0.28.1
45
+ Jinja2==3.1.6
46
+ einops==0.8.2
47
+ mergedeep==1.3.4
48
+ platformdirs==4.10.0
49
+ nvidia-nvjitlink-cu12==12.8.93
50
+ lerobot==0.6.0
51
+ PyYAML==6.0.3
52
+ accelerate==1.14.0
53
+ anyio==4.14.2
54
+ gymnasium==1.3.0
55
+ pip==26.1.2
56
+ huggingface_hub==1.23.0
57
+ diffusers==0.35.2
58
+ annotated-doc==0.0.4
59
+ pyarrow==25.0.0
60
+ python-dateutil==2.9.0.post0
61
+ termcolor==3.3.0
62
+ nvidia-nccl-cu12==2.28.9
63
+ sentry-sdk==2.65.0
64
+ torchcodec==0.11.1
65
+ pyyaml-include==1.4.1
66
+ multiprocess==0.70.19
67
+ wrapt==2.2.2
68
+ typing-inspection==0.4.2
69
+ nvidia-cuda-nvrtc-cu12==12.8.93
70
+ Pygments==2.20.0
71
+ nvidia-cuda-cupti-cu12==12.8.90
72
+ certifi==2026.6.17
73
+ setuptools==78.1.0
74
+ charset-normalizer==3.4.9
75
+ propcache==0.5.2
76
+ multidict==6.7.1
77
+ regex==2026.7.10
78
+ frozenlist==1.8.0
79
+ smmap==5.0.3
80
+ hf-xet==1.5.1
81
+ torchvision==0.26.0+cu128
82
+ gitdb==4.0.12
83
+ transformers==5.5.4
84
+ rich==15.0.0
85
+ cuda-bindings==12.9.4
86
+ protobuf==7.35.1
87
+ aiosignal==1.4.0
88
+ jsonlines==4.0.0
89
+ yarl==1.24.2
90
+ filelock==3.29.0
91
+ av==15.1.0
92
+ networkx==3.6.1
93
+ sympy==1.14.0
94
+ wheel==0.47.0
95
+ nvidia-nvtx-cu12==12.8.90
96
+ xxhash==3.8.1
97
+ toml==0.10.2
98
+ MarkupSafe==3.0.3
99
+ pandas==2.3.3
100
+ safetensors==0.8.0
101
+ click==8.4.2
102
+ fsspec==2026.2.0
103
+ dm-tree==0.1.10
104
+ absl-py==2.5.0
105
+ Farama-Notifications==0.0.6
106
+ mpmath==1.3.0
107
+ importlib_metadata==9.0.0
108
+ markdown-it-py==4.2.0
109
+ h11==0.16.0
110
+ typing-inspect==0.9.0
111
+ httpcore==1.0.9
112
+ annotated-types==0.7.0
113
+ numpy==2.2.6
114
+ nvidia-nvshmem-cu12==3.4.5
115
+ aiohappyeyeballs==2.7.1
116
+ cuda-toolkit==12.8.1
117
+ aiohttp==3.14.1
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/wandb-metadata.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-55-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-07-21T18:14:23.352134Z",
5
+ "args": [
6
+ "--config_path=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json",
7
+ "--resume=true",
8
+ "--dataset.root=/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2",
9
+ "--output_dir=/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K"
10
+ ],
11
+ "program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
12
+ "codePath": "benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
13
+ "codePathLocal": "my_scripts/lerobot_train_yaml.py",
14
+ "git": {
15
+ "remote": "git@github.com:paragon7060/clvla.git",
16
+ "commit": "c1ff610ae4b02827f1c984cef8bbf28a7889bd6f"
17
+ },
18
+ "root": "/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K",
19
+ "host": "ext_minje-box",
20
+ "executable": "/home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python",
21
+ "cpu_count": 112,
22
+ "cpu_count_logical": 224,
23
+ "gpu": "NVIDIA H200",
24
+ "gpu_count": 2,
25
+ "disk": {
26
+ "/": {
27
+ "total": "1888425144320",
28
+ "used": "1061580816384"
29
+ }
30
+ },
31
+ "memory": {
32
+ "total": "2164176568320"
33
+ },
34
+ "gpu_nvidia": [
35
+ {
36
+ "name": "NVIDIA H200",
37
+ "memoryTotal": "150754820096",
38
+ "cudaCores": 16896,
39
+ "architecture": "Hopper",
40
+ "uuid": "GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991"
41
+ },
42
+ {
43
+ "name": "NVIDIA H200",
44
+ "memoryTotal": "150754820096",
45
+ "cudaCores": 16896,
46
+ "architecture": "Hopper",
47
+ "uuid": "GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc"
48
+ }
49
+ ],
50
+ "cudaVersion": "12.8",
51
+ "writerId": "4jpj4uema5zxdg5ph02e1i5khzli4vk4"
52
+ }
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug-core.log ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-07-21T18:14:23.285243333Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpo88oriya/port-19150.txt","pid":19150,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-07-21T18:14:23.28581381Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":19150}
3
+ {"time":"2026-07-21T18:14:23.285809228Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-19150-21411-1232209905/socket","Net":"unix"}}
4
+ {"time":"2026-07-21T18:14:23.38286082Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpbso0rvsy/port-19156.txt","pid":19156,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
5
+ {"time":"2026-07-21T18:14:23.383664879Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":19156}
6
+ {"time":"2026-07-21T18:14:23.383625287Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-19156-21472-1947334501/socket","Net":"unix"}}
7
+ {"time":"2026-07-21T18:14:23.471344728Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
8
+ {"time":"2026-07-21T18:14:23.478360859Z","level":"INFO","msg":"handleInformInit: received","streamId":"ny0tx2de","id":"1(@)"}
9
+ {"time":"2026-07-21T18:14:23.569000336Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
10
+ {"time":"2026-07-21T18:14:23.57555684Z","level":"INFO","msg":"handleInformInit: received","streamId":"bepgb8uu","id":"1(@)"}
11
+ {"time":"2026-07-21T18:14:23.848951854Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"ny0tx2de","id":"1(@)"}
12
+ {"time":"2026-07-21T18:14:24.01659298Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"bepgb8uu","id":"1(@)"}
13
+ {"time":"2026-07-21T18:14:29.505605344Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"r6fec0azzoqo"}
14
+ {"time":"2026-07-21T18:14:30.028386575Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"4nrdajw10sav"}
15
+ {"time":"2026-07-21T18:19:37.712413771Z","level":"INFO","msg":"server: parent process exited, terminating service process"}
16
+ {"time":"2026-07-21T18:19:37.732644068Z","level":"INFO","msg":"server: parent process exited, terminating service process"}
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug.log ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
2
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_setup.py:_flush():81] Configure stats pid to 19156
3
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug.log
5
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug-internal.log
6
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_init.py:init():766] calling init triggers
7
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_init.py:init():771] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'Whalswp/INSIGHTfixposV4_filtered_multispace_v2', 'root': '/home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}, 'affine': {'weight': 1.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec', 'return_uint8': False, 'depth_output_unit': 'mm', 'streaming': False, 'eval_split': 0.0}, 'env': None, 'policy': {'type': 'groot_rkd', 'n_obs_steps': 1, 'input_features': {'observation.state': {'type': <FeatureType.STATE: 'STATE'>, 'shape': [16]}, 'observation.images.wrist': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}, 'observation.images.right_shoulder': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}, 'observation.images.guide': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 224, 224]}}, 'output_features': {'action': {'type': <FeatureType.ACTION: 'ACTION'>, 'shape': [16]}}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model', 'pretrained_revision': None, 'chunk_size': 16, 'n_action_steps': 16, 'max_state_dim': 132, 'max_action_dim': 132, 'normalization_mapping': {'VISUAL': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'STATE': <NormalizationMode.IDENTITY: 'IDENTITY'>, 'ACTION': <NormalizationMode.IDENTITY: 'IDENTITY'>}, 'base_model_path': 'nvidia/GR00T-N1.7-3B', 'action_decode_transform': None, 'embodiment_tag': 'new_embodiment', 'tune_llm': False, 'tune_visual': False, 'tune_projector': True, 'tune_diffusion_model': True, 'tune_vlln': True, 'tune_top_llm_layers': 0, 'num_inference_timesteps': None, 'rtc_ramp_rate': None, 'use_flash_attention': False, 'use_relative_actions': False, 'relative_exclude_joints': [], 'optimizer_lr': 0.0001, 'optimizer_betas': [0.9, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-05, 'warmup_steps': 3000, 'warmup_ratio': 0.05, 'use_bf16': True, 'model_params_fp32': True, 'image_size': [256, 256], 'tokenizer_assets_repo': None, 'lora_rank': 0, 'lora_alpha': 16, 'lora_dropout': 0.1, 'lora_full_model': False, 'video_backend': 'decord', 'balance_dataset_weights': True, 'balance_trajectory_weights': True, 'dataset_paths': None, 'output_dir': './tmp/gr00t', 'save_steps': 1000, 'max_steps': 10000, 'batch_size': 32, 'dataloader_num_workers': 8, 'report_to': 'wandb', 'resume': False, 'rkd_enabled': True, 'rkd_vae_checkpoint': '/home/ext_minje/groot_insight/TimewarpVAE_ckpt/multispace_gripper_dedup_batch_ablation_v1/b512/RD/K08_z16_g4_h256/checkpoint-step-20000-epoch-46.84', 'rkd_student_pooling': 'attention', 'rkd_student_attention_hidden_dim': 512, 'rkd_student_attention_output_dim': 512, 'rkd_fm_loss_weight': 1.0, 'rkd_loss_weight': 1.0, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 2.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K', 'job_name': 'INSIGHT_RD_RKD_TimewarpVAE_RSCLstyle_C60', 'resume': True, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 128, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 60000, 'env_eval_freq': 20000, 'log_freq': 200, 'eval_steps': 0, 'max_eval_samples': 0, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 20000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 0.0001, 'weight_decay': 1e-05, 'grad_clip_norm': 1.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 3000, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': True, 'recording': False, 'recording_repo_id': None, 'recording_private': False}, 'wandb': {'enable': True, 'disable_artifact': True, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': 'bepgb8uu', 'mode': None, 'add_tags': True}, 'peft': None, 'job': {'target': None, 'image': 'huggingface/lerobot-gpu:latest', 'timeout': '2d', 'detach': False, 'tags': []}, 'save_checkpoint_to_hub': False, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000', '_wandb': {}}
9
+ 2026-07-21 18:14:23,353 INFO MainThread:19156 [wandb_init.py:init():814] starting backend
10
+ 2026-07-21 18:14:23,569 INFO MainThread:19156 [wandb_init.py:init():829] sending inform_init request
11
+ 2026-07-21 18:14:24,017 INFO MainThread:19156 [wandb_init.py:init():834] backend started and connected
12
+ 2026-07-21 18:14:24,019 INFO MainThread:19156 [wandb_init.py:init():904] updated telemetry
13
+ 2026-07-21 18:14:24,023 INFO MainThread:19156 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
14
+ 2026-07-21 18:14:24,971 INFO MainThread:19156 [wandb_init.py:init():967] run resumed
15
+ 2026-07-21 18:14:24,973 INFO MainThread:19156 [wandb_init.py:init():972] starting run threads in backend
16
+ 2026-07-21 18:14:25,024 INFO MainThread:19156 [wandb_run.py:_console_start():2523] atexit reg
17
+ 2026-07-21 18:14:25,024 INFO MainThread:19156 [wandb_run.py:_redirect():2373] redirect: wrap_raw
18
+ 2026-07-21 18:14:25,024 INFO MainThread:19156 [wandb_run.py:_redirect():2442] Wrapping output streams.
19
+ 2026-07-21 18:14:25,024 INFO MainThread:19156 [wandb_run.py:_redirect():2465] Redirects installed.
20
+ 2026-07-21 18:14:25,025 INFO MainThread:19156 [wandb_init.py:init():1010] run started, returning control to user process
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_182104-4bcz2tkp/files/output.log ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2026-07-21 18:21:06 db_utils.py:121 Logs will be synced with wandb.
2
+ INFO 2026-07-21 18:21:06 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/4bcz2tkp
3
+ INFO 2026-07-21 18:21:06 ot_train.py:298 Creating dataset
4
+ INFO 2026-07-21 18:21:08 ot_train.py:332 Creating policy
5
+ INFO 2026-07-21 18:21:08 ng_groot.py:191 The Groot policy wraps NVIDIA's GR00T n1.7 model. Loading pretrained model from: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model
6
+ INFO 2026-07-21 18:21:08 ng_groot.py:225 Detected fine-tuned LeRobot checkpoint, loading with state dict...
7
+ Fetching 27 files: 100%|██████████| 27/27 [00:00<00:00, 2481.02it/s]
8
+ `torch_dtype` is deprecated! Use `dtype` instead!
9
+ Loading weights: 100%|██████████| 1031/1031 [00:01<00:00, 623.09it/s]
10
+ Loading weights from local directory
11
+ INFO 2026-07-21 18:21:21 ot_train.py:405 Creating optimizer and scheduler
12
+ INFO 2026-07-21 18:21:24 ot_train.py:439 Output dir: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/restarts/20260719_231636_rerun_save30k60k/RSCLstyle_cosine_60K
13
+ INFO 2026-07-21 18:21:24 ot_train.py:446 cfg.steps=60000 (60K)
14
+ INFO 2026-07-21 18:21:24 ot_train.py:447 dataset.num_frames=218367 (218K)
15
+ INFO 2026-07-21 18:21:24 ot_train.py:448 dataset.num_episodes=4930
16
+ INFO 2026-07-21 18:21:24 ot_train.py:451 Effective batch size: 128 x 1 = 128
17
+ INFO 2026-07-21 18:21:24 ot_train.py:452 num_learnable_params=1622090881 (2B)
18
+ INFO 2026-07-21 18:21:24 ot_train.py:453 num_total_params=3145712033 (3B)
19
+ INFO 2026-07-21 18:21:24 ot_train.py:494 Resuming data order at epoch 26, sample 82688
20
+ Training: 0%| | 0/30000 [00:00<?, ?step/s]INFO 2026-07-21 18:21:24 ot_train.py:604 Start offline training on a fixed dataset, with effective batch size: 128
21
+ Training: 1%| | 200/30000 [08:42<21:01:05, 2.54s/step]INFO 2026-07-21 18:30:07 ot_train.py:649 step:30K smpl:4M ep:87K epch:17.70 loss:0.016 grdn:0.150 lr:5.4e-05 updt_s:1.465 data_s:1.145 smp/s:49 mem_gb:52.98
22
+ Training: 1%|▏ | 400/30000 [17:23<21:14:09, 2.58s/step]INFO 2026-07-21 18:38:48 ot_train.py:649 step:30K smpl:4M ep:88K epch:17.82 loss:0.016 grdn:0.155 lr:5.3e-05 updt_s:1.478 data_s:1.123 smp/s:49 mem_gb:52.98
23
+ Training: 2%|▏ | 600/30000 [26:09<20:31:55, 2.51s/step]INFO 2026-07-21 18:47:33 ot_train.py:649 step:31K smpl:4M ep:88K epch:17.94 loss:0.015 grdn:0.140 lr:5.3e-05 updt_s:1.492 data_s:1.130 smp/s:49 mem_gb:52.91
24
+ Training: 3%|▎ | 800/30000 [34:39<20:26:22, 2.52s/step]INFO 2026-07-21 18:56:03 ot_train.py:649 step:31K smpl:4M ep:89K epch:18.05 loss:0.015 grdn:0.141 lr:5.2e-05 updt_s:1.443 data_s:1.104 smp/s:50 mem_gb:53.00
25
+ Training: 3%|▎ | 1000/30000 [43:02<20:25:45, 2.54s/step]INFO 2026-07-21 19:04:26 ot_train.py:649 step:31K smpl:4M ep:90K epch:18.17 loss:0.015 grdn:0.143 lr:5.2e-05 updt_s:1.416 data_s:1.095 smp/s:51 mem_gb:53.00
26
+ Training: 4%|▍ | 1200/30000 [51:37<20:02:23, 2.50s/step]INFO 2026-07-21 19:13:01 ot_train.py:649 step:31K smpl:4M ep:90K epch:18.29 loss:0.015 grdn:0.140 lr:5.1e-05 updt_s:1.457 data_s:1.113 smp/s:50 mem_gb:53.00
27
+ Training: 5%|▍ | 1400/30000 [1:00:00<19:50:41, 2.50s/step]INFO 2026-07-21 19:21:24 ot_train.py:649 step:31K smpl:4M ep:91K epch:18.41 loss:0.015 grdn:0.138 lr:5.1e-05 updt_s:1.428 data_s:1.083 smp/s:51 mem_gb:53.00
28
+ Training: 5%|▌ | 1600/30000 [1:08:22<18:55:02, 2.40s/step]INFO 2026-07-21 19:29:47 ot_train.py:649 step:32K smpl:4M ep:91K epch:18.52 loss:0.015 grdn:0.139 lr:5.0e-05 updt_s:1.425 data_s:1.085 smp/s:51 mem_gb:53.00
29
+ Training: 6%|▌ | 1800/30000 [1:16:47<19:44:17, 2.52s/step]INFO 2026-07-21 19:38:12 ot_train.py:649 step:32K smpl:4M ep:92K epch:18.64 loss:0.015 grdn:0.136 lr:4.9e-05 updt_s:1.428 data_s:1.093 smp/s:51 mem_gb:52.92
30
+ Training: 7%|▋ | 2000/30000 [1:25:11<19:45:28, 2.54s/step]INFO 2026-07-21 19:46:36 ot_train.py:649 step:32K smpl:4M ep:92K epch:18.76 loss:0.014 grdn:0.136 lr:4.9e-05 updt_s:1.433 data_s:1.082 smp/s:51 mem_gb:53.00
31
+ Training: 7%|▋ | 2200/30000 [1:33:35<19:22:01, 2.51s/step]INFO 2026-07-21 19:54:59 ot_train.py:649 step:32K smpl:4M ep:93K epch:18.87 loss:0.015 grdn:0.146 lr:4.8e-05 updt_s:1.434 data_s:1.080 smp/s:51 mem_gb:53.00
32
+ Training: 8%|▊ | 2400/30000 [1:41:58<19:04:26, 2.49s/step]INFO 2026-07-21 20:03:22 ot_train.py:649 step:32K smpl:4M ep:94K epch:18.99 loss:0.015 grdn:0.129 lr:4.8e-05 updt_s:1.423 data_s:1.088 smp/s:51 mem_gb:53.00
33
+ Training: 9%|▊ | 2600/30000 [1:50:22<19:05:27, 2.51s/step]INFO 2026-07-21 20:11:46 ot_train.py:649 step:33K smpl:4M ep:94K epch:19.11 loss:0.014 grdn:0.136 lr:4.7e-05 updt_s:1.424 data_s:1.093 smp/s:51 mem_gb:53.00
34
+ Training: 9%|▉ | 2800/30000 [1:58:48<19:09:29, 2.54s/step]INFO 2026-07-21 20:20:12 ot_train.py:649 step:33K smpl:4M ep:95K epch:19.23 loss:0.015 grdn:0.149 lr:4.7e-05 updt_s:1.432 data_s:1.094 smp/s:51 mem_gb:52.92
35
+ Training: 10%|█ | 3000/30000 [2:07:15<18:50:33, 2.51s/step]INFO 2026-07-21 20:28:39 ot_train.py:649 step:33K smpl:4M ep:95K epch:19.34 loss:0.014 grdn:0.136 lr:4.6e-05 updt_s:1.440 data_s:1.092 smp/s:51 mem_gb:53.00
36
+ Training: 11%|█ | 3200/30000 [2:15:43<18:37:40, 2.50s/step]INFO 2026-07-21 20:37:07 ot_train.py:649 step:33K smpl:4M ep:96K epch:19.46 loss:0.014 grdn:0.136 lr:4.6e-05 updt_s:1.431 data_s:1.105 smp/s:50 mem_gb:53.00
37
+ Training: 11%|█▏ | 3400/30000 [2:24:11<19:46:03, 2.68s/step]INFO 2026-07-21 20:45:36 ot_train.py:649 step:33K smpl:4M ep:97K epch:19.58 loss:0.014 grdn:0.132 lr:4.5e-05 updt_s:1.434 data_s:1.104 smp/s:50 mem_gb:53.00
38
+ Training: 12%|█▏ | 3600/30000 [2:32:40<18:36:07, 2.54s/step]INFO 2026-07-21 20:54:05 ot_train.py:649 step:34K smpl:4M ep:97K epch:19.70 loss:0.014 grdn:0.143 lr:4.4e-05 updt_s:1.427 data_s:1.115 smp/s:50 mem_gb:53.00
39
+ Training: 13%|█▎ | 3800/30000 [2:41:06<18:29:16, 2.54s/step]INFO 2026-07-21 21:02:30 ot_train.py:649 step:34K smpl:4M ep:98K epch:19.81 loss:0.014 grdn:0.132 lr:4.4e-05 updt_s:1.422 data_s:1.100 smp/s:51 mem_gb:53.00
40
+ Training: 13%|█▎ | 4000/30000 [2:49:34<18:16:02, 2.53s/step]INFO 2026-07-21 21:10:58 ot_train.py:649 step:34K smpl:4M ep:98K epch:19.93 loss:0.013 grdn:0.128 lr:4.3e-05 updt_s:1.427 data_s:1.110 smp/s:50 mem_gb:52.92
41
+ Training: 14%|█▍ | 4200/30000 [2:57:59<17:52:33, 2.49s/step]INFO 2026-07-21 21:19:23 ot_train.py:649 step:34K smpl:4M ep:99K epch:20.05 loss:0.013 grdn:0.127 lr:4.3e-05 updt_s:1.419 data_s:1.102 smp/s:51 mem_gb:53.00
42
+ Training: 15%|█▍ | 4400/30000 [3:06:26<18:23:18, 2.59s/step]INFO 2026-07-21 21:27:51 ot_train.py:649 step:34K smpl:4M ep:99K epch:20.16 loss:0.012 grdn:0.120 lr:4.2e-05 updt_s:1.418 data_s:1.117 smp/s:50 mem_gb:53.00
43
+ Training: 15%|█▌ | 4600/30000 [3:14:54<18:02:58, 2.56s/step]INFO 2026-07-21 21:36:19 ot_train.py:649 step:35K smpl:4M ep:100K epch:20.28 loss:0.013 grdn:0.133 lr:4.2e-05 updt_s:1.409 data_s:1.126 smp/s:51 mem_gb:53.00
44
+ Training: 16%|█▌ | 4800/30000 [3:23:21<17:47:11, 2.54s/step]INFO 2026-07-21 21:44:45 ot_train.py:649 step:35K smpl:4M ep:101K epch:20.40 loss:0.012 grdn:0.121 lr:4.1e-05 updt_s:1.413 data_s:1.116 smp/s:51 mem_gb:53.00
45
+ Training: 17%|█▋ | 5000/30000 [3:31:57<22:14:24, 3.20s/step]INFO 2026-07-21 21:53:21 ot_train.py:649 step:35K smpl:4M ep:101K epch:20.52 loss:0.012 grdn:0.122 lr:4.1e-05 updt_s:1.433 data_s:1.144 smp/s:50 mem_gb:52.92
46
+ Training: 17%|█▋ | 5200/30000 [3:40:36<17:10:12, 2.49s/step]INFO 2026-07-21 22:02:00 ot_train.py:649 step:35K smpl:5M ep:102K epch:20.63 loss:0.012 grdn:0.124 lr:4.0e-05 updt_s:1.449 data_s:1.142 smp/s:49 mem_gb:53.00
47
+ Training: 18%|█▊ | 5400/30000 [3:49:00<17:14:15, 2.52s/step]INFO 2026-07-21 22:10:25 ot_train.py:649 step:35K smpl:5M ep:102K epch:20.75 loss:0.012 grdn:0.127 lr:4.0e-05 updt_s:1.402 data_s:1.118 smp/s:51 mem_gb:53.00
48
+ Training: 19%|█▊ | 5600/30000 [3:57:25<17:27:47, 2.58s/step]INFO 2026-07-21 22:18:50 ot_train.py:649 step:36K smpl:5M ep:103K epch:20.87 loss:0.012 grdn:0.123 lr:3.9e-05 updt_s:1.401 data_s:1.118 smp/s:51 mem_gb:53.00
49
+ Training: 19%|█▉ | 5800/30000 [4:05:54<16:42:34, 2.49s/step]INFO 2026-07-21 22:27:18 ot_train.py:649 step:36K smpl:5M ep:103K epch:20.98 loss:0.012 grdn:0.118 lr:3.9e-05 updt_s:1.418 data_s:1.121 smp/s:50 mem_gb:53.00
50
+ Training: 20%|██ | 6000/30000 [4:14:24<17:07:39, 2.57s/step]INFO 2026-07-21 22:35:49 ot_train.py:649 step:36K smpl:5M ep:104K epch:21.10 loss:0.012 grdn:0.122 lr:3.8e-05 updt_s:1.437 data_s:1.113 smp/s:50 mem_gb:53.00
51
+ Training: 21%|██ | 6200/30000 [4:22:57<16:45:36, 2.54s/step]INFO 2026-07-21 22:44:22 ot_train.py:649 step:36K smpl:5M ep:105K epch:21.22 loss:0.012 grdn:0.119 lr:3.7e-05 updt_s:1.432 data_s:1.129 smp/s:50 mem_gb:52.92
52
+ Training: 21%|██▏ | 6400/30000 [4:31:30<16:56:49, 2.59s/step]INFO 2026-07-21 22:52:54 ot_train.py:649 step:36K smpl:5M ep:105K epch:21.34 loss:0.011 grdn:0.123 lr:3.7e-05 updt_s:1.422 data_s:1.137 smp/s:50 mem_gb:53.00
53
+ Training: 22%|██▏ | 6600/30000 [4:39:59<16:19:07, 2.51s/step]INFO 2026-07-21 23:01:24 ot_train.py:649 step:37K smpl:5M ep:106K epch:21.45 loss:0.012 grdn:0.127 lr:3.6e-05 updt_s:1.406 data_s:1.136 smp/s:50 mem_gb:53.00
54
+ Training: 23%|██▎ | 6800/30000 [4:48:37<16:48:20, 2.61s/step]INFO 2026-07-21 23:10:01 ot_train.py:649 step:37K smpl:5M ep:106K epch:21.57 loss:0.012 grdn:0.121 lr:3.6e-05 updt_s:1.438 data_s:1.145 smp/s:50 mem_gb:53.00
55
+ Training: 23%|██▎ | 7000/30000 [4:57:16<16:30:44, 2.58s/step]INFO 2026-07-21 23:18:41 ot_train.py:649 step:37K smpl:5M ep:107K epch:21.69 loss:0.012 grdn:0.124 lr:3.5e-05 updt_s:1.433 data_s:1.162 smp/s:49 mem_gb:53.00
56
+ Training: 24%|██▍ | 7200/30000 [5:05:51<16:51:35, 2.66s/step]INFO 2026-07-21 23:27:15 ot_train.py:649 step:37K smpl:5M ep:108K epch:21.81 loss:0.011 grdn:0.118 lr:3.5e-05 updt_s:1.429 data_s:1.138 smp/s:50 mem_gb:53.00
57
+ Training: 25%|██▍ | 7400/30000 [5:14:40<17:17:12, 2.75s/step]INFO 2026-07-21 23:36:05 ot_train.py:649 step:37K smpl:5M ep:108K epch:21.92 loss:0.011 grdn:0.117 lr:3.4e-05 updt_s:1.464 data_s:1.181 smp/s:48 mem_gb:52.92
58
+ Training: 25%|██▌ | 7600/30000 [5:23:24<16:05:08, 2.59s/step]INFO 2026-07-21 23:44:48 ot_train.py:649 step:38K smpl:5M ep:109K epch:22.04 loss:0.011 grdn:0.112 lr:3.4e-05 updt_s:1.504 data_s:1.109 smp/s:49 mem_gb:53.00
59
+ Training: 26%|██▌ | 7800/30000 [5:32:04<15:46:19, 2.56s/step]INFO 2026-07-21 23:53:28 ot_train.py:649 step:38K smpl:5M ep:109K epch:22.16 loss:0.011 grdn:0.111 lr:3.3e-05 updt_s:1.530 data_s:1.065 smp/s:49 mem_gb:53.00
60
+ Training: 27%|██▋ | 8000/30000 [5:40:35<15:13:19, 2.49s/step]INFO 2026-07-22 00:02:00 ot_train.py:649 step:38K smpl:5M ep:110K epch:22.27 loss:0.010 grdn:0.115 lr:3.3e-05 updt_s:1.511 data_s:1.044 smp/s:50 mem_gb:53.00
61
+ Training: 27%|██▋ | 8200/30000 [5:49:24<15:58:12, 2.64s/step]INFO 2026-07-22 00:10:49 ot_train.py:649 step:38K smpl:5M ep:110K epch:22.39 loss:0.011 grdn:0.117 lr:3.2e-05 updt_s:1.494 data_s:1.146 smp/s:48 mem_gb:53.00
62
+ Training: 28%|██▊ | 8400/30000 [5:58:20<15:33:58, 2.59s/step]INFO 2026-07-22 00:19:45 ot_train.py:649 step:38K smpl:5M ep:111K epch:22.51 loss:0.011 grdn:0.125 lr:3.2e-05 updt_s:1.512 data_s:1.162 smp/s:48 mem_gb:52.92
63
+ Training: 29%|██▊ | 8600/30000 [6:07:30<15:14:41, 2.56s/step]INFO 2026-07-22 00:28:55 ot_train.py:649 step:39K smpl:5M ep:112K epch:22.63 loss:0.010 grdn:0.113 lr:3.1e-05 updt_s:1.574 data_s:1.171 smp/s:47 mem_gb:53.00
64
+ Training: 29%|██▉ | 8800/30000 [6:16:20<15:24:46, 2.62s/step]INFO 2026-07-22 00:37:44 ot_train.py:649 step:39K smpl:5M ep:112K epch:22.74 loss:0.010 grdn:0.110 lr:3.1e-05 updt_s:1.486 data_s:1.158 smp/s:48 mem_gb:53.00
65
+ Training: 30%|███ | 9000/30000 [6:27:46<20:42:24, 3.55s/step]INFO 2026-07-22 00:49:10 ot_train.py:649 step:39K smpl:5M ep:113K epch:22.86 loss:0.010 grdn:0.113 lr:3.0e-05 updt_s:1.788 data_s:1.636 smp/s:37 mem_gb:53.00
66
+ Training: 31%|███ | 9200/30000 [6:38:21<15:36:56, 2.70s/step]INFO 2026-07-22 00:59:46 ot_train.py:649 step:39K smpl:5M ep:113K epch:22.98 loss:0.010 grdn:0.116 lr:3.0e-05 updt_s:1.685 data_s:1.485 smp/s:40 mem_gb:53.00
67
+ Training: 31%|███▏ | 9400/30000 [6:47:21<15:47:46, 2.76s/step]INFO 2026-07-22 01:08:45 ot_train.py:649 step:39K smpl:5M ep:114K epch:23.10 loss:0.010 grdn:0.116 lr:2.9e-05 updt_s:1.521 data_s:1.173 smp/s:48 mem_gb:53.00
68
+ Training: 32%|███▏ | 9600/30000 [6:56:31<15:06:50, 2.67s/step]INFO 2026-07-22 01:17:56 ot_train.py:649 step:40K smpl:5M ep:114K epch:23.21 loss:0.010 grdn:0.115 lr:2.9e-05 updt_s:1.533 data_s:1.214 smp/s:47 mem_gb:52.92
69
+ Training: 33%|███▎ | 9800/30000 [7:05:21<15:21:01, 2.74s/step]INFO 2026-07-22 01:26:46 ot_train.py:649 step:40K smpl:5M ep:115K epch:23.33 loss:0.009 grdn:0.111 lr:2.8e-05 updt_s:1.479 data_s:1.168 smp/s:48 mem_gb:53.00
70
+ Training: 33%|███▎ | 10000/30000 [7:14:26<15:05:25, 2.72s/step]INFO 2026-07-22 01:35:51 ot_train.py:649 step:40K smpl:5M ep:116K epch:23.45 loss:0.010 grdn:0.108 lr:2.8e-05 updt_s:1.506 data_s:1.214 smp/s:47 mem_gb:53.00
71
+ Training: 34%|███▍ | 10200/30000 [7:23:09<14:23:22, 2.62s/step]INFO 2026-07-22 01:44:34 ot_train.py:649 step:40K smpl:5M ep:116K epch:23.56 loss:0.010 grdn:0.117 lr:2.7e-05 updt_s:1.449 data_s:1.162 smp/s:49 mem_gb:53.00
72
+ Training: 34%|███▍ | 10347/30000 [7:29:53<14:23:55, 2.64s/step]
Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_182104-4bcz2tkp/files/requirements.txt ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ tzdata==2026.3
2
+ nvidia-cufile-cu12==1.13.1.3
3
+ wandb==0.27.2
4
+ six==1.17.0
5
+ peft==0.19.1
6
+ nvidia-curand-cu12==10.3.9.90
7
+ nvidia-cuda-runtime-cu12==12.8.90
8
+ datasets==4.8.5
9
+ attrs==26.1.0
10
+ cloudpickle==3.1.2
11
+ packaging==25.0
12
+ nvidia-cublas-cu12==12.8.4.1
13
+ cmake==4.1.3
14
+ psutil==7.2.2
15
+ pydantic==2.13.4
16
+ GitPython==3.1.51
17
+ pillow==12.2.0
18
+ pytz==2026.2
19
+ cuda-pathfinder==1.2.2
20
+ nvidia-cusparse-cu12==12.5.8.93
21
+ typing_extensions==4.15.0
22
+ urllib3==2.7.0
23
+ pydantic_core==2.46.4
24
+ draccus==0.10.0
25
+ mypy_extensions==1.1.0
26
+ dill==0.4.1
27
+ typer==0.26.8
28
+ decord==0.6.0
29
+ tqdm==4.68.4
30
+ triton==3.6.0
31
+ idna==3.18
32
+ requests==2.34.2
33
+ nvidia-cudnn-cu12==9.19.0.56
34
+ mdurl==0.1.2
35
+ shellingham==1.5.4
36
+ zipp==4.1.0
37
+ timm==1.0.28
38
+ torch==2.11.0+cu128
39
+ tokenizers==0.22.2
40
+ nvidia-cusparselt-cu12==0.7.1
41
+ nvidia-cufft-cu12==11.3.3.83
42
+ nvidia-cusolver-cu12==11.7.3.90
43
+ opencv-python-headless==4.13.0.92
44
+ httpx==0.28.1
45
+ Jinja2==3.1.6
46
+ einops==0.8.2
47
+ mergedeep==1.3.4
48
+ platformdirs==4.10.0
49
+ nvidia-nvjitlink-cu12==12.8.93
50
+ lerobot==0.6.0
51
+ PyYAML==6.0.3
52
+ accelerate==1.14.0
53
+ anyio==4.14.2
54
+ gymnasium==1.3.0
55
+ pip==26.1.2
56
+ huggingface_hub==1.23.0
57
+ diffusers==0.35.2
58
+ annotated-doc==0.0.4
59
+ pyarrow==25.0.0
60
+ python-dateutil==2.9.0.post0
61
+ termcolor==3.3.0
62
+ nvidia-nccl-cu12==2.28.9
63
+ sentry-sdk==2.65.0
64
+ torchcodec==0.11.1
65
+ pyyaml-include==1.4.1
66
+ multiprocess==0.70.19
67
+ wrapt==2.2.2
68
+ typing-inspection==0.4.2
69
+ nvidia-cuda-nvrtc-cu12==12.8.93
70
+ Pygments==2.20.0
71
+ nvidia-cuda-cupti-cu12==12.8.90
72
+ certifi==2026.6.17
73
+ setuptools==78.1.0
74
+ charset-normalizer==3.4.9
75
+ propcache==0.5.2
76
+ multidict==6.7.1
77
+ regex==2026.7.10
78
+ frozenlist==1.8.0
79
+ smmap==5.0.3
80
+ hf-xet==1.5.1
81
+ torchvision==0.26.0+cu128
82
+ gitdb==4.0.12
83
+ transformers==5.5.4
84
+ rich==15.0.0
85
+ cuda-bindings==12.9.4
86
+ protobuf==7.35.1
87
+ aiosignal==1.4.0
88
+ jsonlines==4.0.0
89
+ yarl==1.24.2
90
+ filelock==3.29.0
91
+ av==15.1.0
92
+ networkx==3.6.1
93
+ sympy==1.14.0
94
+ wheel==0.47.0
95
+ nvidia-nvtx-cu12==12.8.90
96
+ xxhash==3.8.1
97
+ toml==0.10.2
98
+ MarkupSafe==3.0.3
99
+ pandas==2.3.3
100
+ safetensors==0.8.0
101
+ click==8.4.2
102
+ fsspec==2026.2.0
103
+ dm-tree==0.1.10
104
+ absl-py==2.5.0
105
+ Farama-Notifications==0.0.6
106
+ mpmath==1.3.0
107
+ importlib_metadata==9.0.0
108
+ markdown-it-py==4.2.0
109
+ h11==0.16.0
110
+ typing-inspect==0.9.0
111
+ httpcore==1.0.9
112
+ annotated-types==0.7.0
113
+ numpy==2.2.6
114
+ nvidia-nvshmem-cu12==3.4.5
115
+ aiohappyeyeballs==2.7.1
116
+ cuda-toolkit==12.8.1
117
+ aiohttp==3.14.1