Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/action_space_manifest.json +205 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/action_space_manifest.json +205 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/config.json +107 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/policy_preprocessor.json +78 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/prompt_manifest.json +40 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/pretrained_model/train_config.json +258 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/prompt_manifest.json +40 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/optimizer_param_groups.json +577 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/scheduler_state.json +18 -0
- Abs_6D + DeltaEE/Baseline/checkpoints/030000/training_state/training_step.json +5 -0
- Abs_6D + DeltaEE/Baseline/wandb/debug-internal.log.rerun_20260719_231636 +0 -0
- Abs_6D + DeltaEE/Baseline/wandb/debug.log.rerun_20260719_231636 +19 -0
- Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/files/wandb-metadata.json +91 -0
- Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug-internal.log +0 -0
- Abs_6D + DeltaEE/Baseline/wandb/run-20260719_231648-h160hqm2/logs/debug.log +19 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/phase_schedule.json +55 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/action_space_manifest.json +205 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/config.json +117 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/phase_schedule.json +55 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/policy_preprocessor.json +82 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/pretrained_model/train_config.json +268 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/source_config.yaml +65 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/optimizer_param_groups.json +585 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/checkpoints/030000/training_state/scheduler_state.json +18 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_175602-r3cz38bm/logs/debug-internal.log +18 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/LAPstyle_linear_6K/wandb/run-20260721_181953-a9vyia29/logs/debug-internal.log +15 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/requirements.txt +117 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260719_231652-bepgb8uu/files/wandb-metadata.json +91 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/config.yaml +333 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/requirements.txt +117 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-metadata.json +52 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/files/wandb-summary.json +1 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-core.log +45 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_175602-bepgb8uu/logs/debug-internal.log +18 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/config.yaml +332 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/files/wandb-metadata.json +52 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180257-bepgb8uu/logs/debug-core.log +48 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/config.yaml +332 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/output.log +33 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/files/wandb-summary.json +1 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_180758-bepgb8uu/logs/debug.log +46 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/output.log +21 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/requirements.txt +117 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/files/wandb-metadata.json +52 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug-core.log +16 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_181423-bepgb8uu/logs/debug.log +20 -0
- Abs_6D + DeltaEE/RKD_TimewarpVAE/RSCLstyle_cosine_60K/wandb/run-20260721_182104-4bcz2tkp/files/output.log +72 -0
- 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
|