Add files using upload-large-folder tool
Browse files- Abs_6D + DeltaEE/Baseline/resolved_config.yaml +24 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/pretrained_model/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/pretrained_model/train_config.json +269 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/resolved_config.yaml +26 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/optimizer_param_groups.json +241 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/scheduler_state.json +18 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/training_step.json +5 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/config.json +118 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/policy_preprocessor.json +78 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/train_config.json +269 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/prompt_manifest.json +40 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/training_state/optimizer_param_groups.json +241 -0
- Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/training_state/training_step.json +5 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/debug.log +20 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/files/output.log +231 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/files/requirements.txt +117 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug-core.log +14 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug-internal.log +0 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug.log +19 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/output.log +70 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/wandb-metadata.json +50 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/wandb-summary.json +1 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug-core.log +62 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug-internal.log +185 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug.log +46 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/files/output.log +41 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/files/requirements.txt +117 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug.log +20 -0
- Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/run-21sn30p5.wandb +0 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/config.json +118 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/phase_schedule.json +32 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/policy_preprocessor.json +82 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/prompt_manifest.json +40 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/train_config.json +269 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/training_state/optimizer_param_groups.json +249 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/training_state/scheduler_state.json +18 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/action_space_manifest.json +144 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/policy_postprocessor.json +23 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json +269 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/training_state/optimizer_param_groups.json +249 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/training_state/training_step.json +5 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/wandb/run-20260722_020800-rd8tz18f/files/output.log +226 -0
- Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/wandb/run-20260722_020800-rd8tz18f/logs/debug-internal.log +0 -0
Abs_6D + DeltaEE/Baseline/resolved_config.yaml
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
dataset:
|
| 2 |
+
repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
|
| 3 |
+
root: /home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
|
| 4 |
+
policy:
|
| 5 |
+
type: groot
|
| 6 |
+
device: cuda
|
| 7 |
+
chunk_size: 16
|
| 8 |
+
n_action_steps: 16
|
| 9 |
+
push_to_hub: false
|
| 10 |
+
tune_llm: false
|
| 11 |
+
tune_visual: false
|
| 12 |
+
tune_top_llm_layers: 0
|
| 13 |
+
tune_projector: true
|
| 14 |
+
tune_diffusion_model: true
|
| 15 |
+
tune_vlln: true
|
| 16 |
+
seed: 42
|
| 17 |
+
batch_size: 128
|
| 18 |
+
steps: 60000
|
| 19 |
+
log_freq: 200
|
| 20 |
+
output_dir: /home/ext_minje/groot_insight/train_ckpt/Abs_6D + DeltaEE/Baseline
|
| 21 |
+
job_name: INSIGHT_RD_baseline
|
| 22 |
+
wandb:
|
| 23 |
+
enable: true
|
| 24 |
+
disable_artifact: true
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/pretrained_model/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/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": 10,
|
| 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/DiT_Layer/8/Baseline/checkpoints/010000/pretrained_model/train_config.json
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": null,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 194 |
+
"rkd_enabled": false,
|
| 195 |
+
"rkd_vae_checkpoint": null,
|
| 196 |
+
"rkd_student_pooling": "attention",
|
| 197 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 198 |
+
"rkd_student_attention_output_dim": 512,
|
| 199 |
+
"rkd_fm_loss_weight": 1.0,
|
| 200 |
+
"rkd_loss_weight": 0.1,
|
| 201 |
+
"rkd_distance_weight": 1.0,
|
| 202 |
+
"rkd_angle_weight": 1.0
|
| 203 |
+
},
|
| 204 |
+
"reward_model": null,
|
| 205 |
+
"output_dir": "/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline",
|
| 206 |
+
"job_name": "INSIGHT_6D_DiT8_Baseline",
|
| 207 |
+
"resume": false,
|
| 208 |
+
"seed": 42,
|
| 209 |
+
"cudnn_deterministic": false,
|
| 210 |
+
"num_workers": 4,
|
| 211 |
+
"batch_size": 64,
|
| 212 |
+
"prefetch_factor": 4,
|
| 213 |
+
"persistent_workers": true,
|
| 214 |
+
"steps": 60000,
|
| 215 |
+
"env_eval_freq": 20000,
|
| 216 |
+
"log_freq": 200,
|
| 217 |
+
"eval_steps": 0,
|
| 218 |
+
"max_eval_samples": 0,
|
| 219 |
+
"tolerance_s": 0.0001,
|
| 220 |
+
"save_checkpoint": true,
|
| 221 |
+
"save_freq": 20000,
|
| 222 |
+
"use_policy_training_preset": true,
|
| 223 |
+
"optimizer": {
|
| 224 |
+
"type": "adamw",
|
| 225 |
+
"lr": 0.0001,
|
| 226 |
+
"weight_decay": 1e-05,
|
| 227 |
+
"grad_clip_norm": 1.0,
|
| 228 |
+
"betas": [
|
| 229 |
+
0.9,
|
| 230 |
+
0.999
|
| 231 |
+
],
|
| 232 |
+
"eps": 1e-08
|
| 233 |
+
},
|
| 234 |
+
"scheduler": {
|
| 235 |
+
"type": "diffuser",
|
| 236 |
+
"num_warmup_steps": 500,
|
| 237 |
+
"name": "cosine"
|
| 238 |
+
},
|
| 239 |
+
"eval": {
|
| 240 |
+
"n_episodes": 50,
|
| 241 |
+
"batch_size": 50,
|
| 242 |
+
"use_async_envs": true,
|
| 243 |
+
"recording": false,
|
| 244 |
+
"recording_repo_id": null,
|
| 245 |
+
"recording_private": false
|
| 246 |
+
},
|
| 247 |
+
"wandb": {
|
| 248 |
+
"enable": true,
|
| 249 |
+
"disable_artifact": true,
|
| 250 |
+
"project": "lerobot",
|
| 251 |
+
"entity": null,
|
| 252 |
+
"notes": null,
|
| 253 |
+
"run_id": "21sn30p5",
|
| 254 |
+
"mode": null,
|
| 255 |
+
"add_tags": true
|
| 256 |
+
},
|
| 257 |
+
"peft": null,
|
| 258 |
+
"job": {
|
| 259 |
+
"target": null,
|
| 260 |
+
"image": "huggingface/lerobot-gpu:latest",
|
| 261 |
+
"timeout": "2d",
|
| 262 |
+
"detach": false,
|
| 263 |
+
"tags": []
|
| 264 |
+
},
|
| 265 |
+
"save_checkpoint_to_hub": false,
|
| 266 |
+
"sample_weighting": null,
|
| 267 |
+
"rename_map": {},
|
| 268 |
+
"checkpoint_path": null
|
| 269 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/resolved_config.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
dataset:
|
| 2 |
+
repo_id: Whalswp/INSIGHTfixposV4_filtered_multispace_v2
|
| 3 |
+
root: /home/ext_minje/INSIGHTfixposV4_filtered_multispace_v2
|
| 4 |
+
policy:
|
| 5 |
+
type: groot_rkd
|
| 6 |
+
device: cuda
|
| 7 |
+
chunk_size: 16
|
| 8 |
+
n_action_steps: 16
|
| 9 |
+
push_to_hub: false
|
| 10 |
+
tune_llm: false
|
| 11 |
+
tune_visual: false
|
| 12 |
+
tune_top_llm_layers: 0
|
| 13 |
+
action_expert_num_layers: 8
|
| 14 |
+
tune_projector: true
|
| 15 |
+
tune_diffusion_model: true
|
| 16 |
+
tune_vlln: true
|
| 17 |
+
rkd_enabled: false
|
| 18 |
+
seed: 42
|
| 19 |
+
batch_size: 64
|
| 20 |
+
steps: 60000
|
| 21 |
+
log_freq: 200
|
| 22 |
+
output_dir: /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline
|
| 23 |
+
job_name: INSIGHT_6D_DiT8_Baseline
|
| 24 |
+
wandb:
|
| 25 |
+
enable: true
|
| 26 |
+
disable_artifact: true
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/optimizer_param_groups.json
ADDED
|
@@ -0,0 +1,241 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"lr": 9.384074610206495e-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 |
+
]
|
| 119 |
+
},
|
| 120 |
+
{
|
| 121 |
+
"weight_decay": 0.0,
|
| 122 |
+
"lr": 9.384074610206495e-05,
|
| 123 |
+
"betas": [
|
| 124 |
+
0.9,
|
| 125 |
+
0.999
|
| 126 |
+
],
|
| 127 |
+
"eps": 1e-08,
|
| 128 |
+
"amsgrad": false,
|
| 129 |
+
"maximize": false,
|
| 130 |
+
"foreach": null,
|
| 131 |
+
"capturable": false,
|
| 132 |
+
"differentiable": false,
|
| 133 |
+
"fused": null,
|
| 134 |
+
"decoupled_weight_decay": true,
|
| 135 |
+
"initial_lr": 0.0001,
|
| 136 |
+
"params": [
|
| 137 |
+
99,
|
| 138 |
+
100,
|
| 139 |
+
101,
|
| 140 |
+
102,
|
| 141 |
+
103,
|
| 142 |
+
104,
|
| 143 |
+
105,
|
| 144 |
+
106,
|
| 145 |
+
107,
|
| 146 |
+
108,
|
| 147 |
+
109,
|
| 148 |
+
110,
|
| 149 |
+
111,
|
| 150 |
+
112,
|
| 151 |
+
113,
|
| 152 |
+
114,
|
| 153 |
+
115,
|
| 154 |
+
116,
|
| 155 |
+
117,
|
| 156 |
+
118,
|
| 157 |
+
119,
|
| 158 |
+
120,
|
| 159 |
+
121,
|
| 160 |
+
122,
|
| 161 |
+
123,
|
| 162 |
+
124,
|
| 163 |
+
125,
|
| 164 |
+
126,
|
| 165 |
+
127,
|
| 166 |
+
128,
|
| 167 |
+
129,
|
| 168 |
+
130,
|
| 169 |
+
131,
|
| 170 |
+
132,
|
| 171 |
+
133,
|
| 172 |
+
134,
|
| 173 |
+
135,
|
| 174 |
+
136,
|
| 175 |
+
137,
|
| 176 |
+
138,
|
| 177 |
+
139,
|
| 178 |
+
140,
|
| 179 |
+
141,
|
| 180 |
+
142,
|
| 181 |
+
143,
|
| 182 |
+
144,
|
| 183 |
+
145,
|
| 184 |
+
146,
|
| 185 |
+
147,
|
| 186 |
+
148,
|
| 187 |
+
149,
|
| 188 |
+
150,
|
| 189 |
+
151,
|
| 190 |
+
152,
|
| 191 |
+
153,
|
| 192 |
+
154,
|
| 193 |
+
155,
|
| 194 |
+
156,
|
| 195 |
+
157,
|
| 196 |
+
158,
|
| 197 |
+
159,
|
| 198 |
+
160,
|
| 199 |
+
161,
|
| 200 |
+
162,
|
| 201 |
+
163,
|
| 202 |
+
164,
|
| 203 |
+
165,
|
| 204 |
+
166,
|
| 205 |
+
167,
|
| 206 |
+
168,
|
| 207 |
+
169,
|
| 208 |
+
170,
|
| 209 |
+
171,
|
| 210 |
+
172,
|
| 211 |
+
173,
|
| 212 |
+
174,
|
| 213 |
+
175,
|
| 214 |
+
176,
|
| 215 |
+
177,
|
| 216 |
+
178,
|
| 217 |
+
179,
|
| 218 |
+
180,
|
| 219 |
+
181,
|
| 220 |
+
182,
|
| 221 |
+
183,
|
| 222 |
+
184,
|
| 223 |
+
185,
|
| 224 |
+
186,
|
| 225 |
+
187,
|
| 226 |
+
188,
|
| 227 |
+
189,
|
| 228 |
+
190,
|
| 229 |
+
191,
|
| 230 |
+
192,
|
| 231 |
+
193,
|
| 232 |
+
194,
|
| 233 |
+
195,
|
| 234 |
+
196,
|
| 235 |
+
197,
|
| 236 |
+
198,
|
| 237 |
+
199,
|
| 238 |
+
200
|
| 239 |
+
]
|
| 240 |
+
}
|
| 241 |
+
]
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/scheduler_state.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"base_lrs": [
|
| 3 |
+
0.0001,
|
| 4 |
+
0.0001
|
| 5 |
+
],
|
| 6 |
+
"last_epoch": 10000,
|
| 7 |
+
"_step_count": 10001,
|
| 8 |
+
"_is_initial": false,
|
| 9 |
+
"_get_lr_called_within_step": false,
|
| 10 |
+
"_last_lr": [
|
| 11 |
+
9.384074610206495e-05,
|
| 12 |
+
9.384074610206495e-05
|
| 13 |
+
],
|
| 14 |
+
"lr_lambdas": [
|
| 15 |
+
null,
|
| 16 |
+
null
|
| 17 |
+
]
|
| 18 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000/training_state/training_step.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"step": 10000,
|
| 3 |
+
"num_processes": 1,
|
| 4 |
+
"batch_size": 64
|
| 5 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/config.json
ADDED
|
@@ -0,0 +1,118 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": null,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 109 |
+
"rkd_enabled": false,
|
| 110 |
+
"rkd_vae_checkpoint": null,
|
| 111 |
+
"rkd_student_pooling": "attention",
|
| 112 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 113 |
+
"rkd_student_attention_output_dim": 512,
|
| 114 |
+
"rkd_fm_loss_weight": 1.0,
|
| 115 |
+
"rkd_loss_weight": 0.1,
|
| 116 |
+
"rkd_distance_weight": 1.0,
|
| 117 |
+
"rkd_angle_weight": 1.0
|
| 118 |
+
}
|
Abs_6D/DiT_Layer/8/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": 10,
|
| 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/DiT_Layer/8/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/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/train_config.json
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": null,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 194 |
+
"rkd_enabled": false,
|
| 195 |
+
"rkd_vae_checkpoint": null,
|
| 196 |
+
"rkd_student_pooling": "attention",
|
| 197 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 198 |
+
"rkd_student_attention_output_dim": 512,
|
| 199 |
+
"rkd_fm_loss_weight": 1.0,
|
| 200 |
+
"rkd_loss_weight": 0.1,
|
| 201 |
+
"rkd_distance_weight": 1.0,
|
| 202 |
+
"rkd_angle_weight": 1.0
|
| 203 |
+
},
|
| 204 |
+
"reward_model": null,
|
| 205 |
+
"output_dir": "/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline",
|
| 206 |
+
"job_name": "INSIGHT_6D_DiT8_Baseline",
|
| 207 |
+
"resume": false,
|
| 208 |
+
"seed": 42,
|
| 209 |
+
"cudnn_deterministic": false,
|
| 210 |
+
"num_workers": 4,
|
| 211 |
+
"batch_size": 64,
|
| 212 |
+
"prefetch_factor": 4,
|
| 213 |
+
"persistent_workers": true,
|
| 214 |
+
"steps": 60000,
|
| 215 |
+
"env_eval_freq": 20000,
|
| 216 |
+
"log_freq": 200,
|
| 217 |
+
"eval_steps": 0,
|
| 218 |
+
"max_eval_samples": 0,
|
| 219 |
+
"tolerance_s": 0.0001,
|
| 220 |
+
"save_checkpoint": true,
|
| 221 |
+
"save_freq": 20000,
|
| 222 |
+
"use_policy_training_preset": true,
|
| 223 |
+
"optimizer": {
|
| 224 |
+
"type": "adamw",
|
| 225 |
+
"lr": 0.0001,
|
| 226 |
+
"weight_decay": 1e-05,
|
| 227 |
+
"grad_clip_norm": 1.0,
|
| 228 |
+
"betas": [
|
| 229 |
+
0.9,
|
| 230 |
+
0.999
|
| 231 |
+
],
|
| 232 |
+
"eps": 1e-08
|
| 233 |
+
},
|
| 234 |
+
"scheduler": {
|
| 235 |
+
"type": "diffuser",
|
| 236 |
+
"num_warmup_steps": 500,
|
| 237 |
+
"name": "cosine"
|
| 238 |
+
},
|
| 239 |
+
"eval": {
|
| 240 |
+
"n_episodes": 50,
|
| 241 |
+
"batch_size": 50,
|
| 242 |
+
"use_async_envs": true,
|
| 243 |
+
"recording": false,
|
| 244 |
+
"recording_repo_id": null,
|
| 245 |
+
"recording_private": false
|
| 246 |
+
},
|
| 247 |
+
"wandb": {
|
| 248 |
+
"enable": true,
|
| 249 |
+
"disable_artifact": true,
|
| 250 |
+
"project": "lerobot",
|
| 251 |
+
"entity": null,
|
| 252 |
+
"notes": null,
|
| 253 |
+
"run_id": "21sn30p5",
|
| 254 |
+
"mode": null,
|
| 255 |
+
"add_tags": true
|
| 256 |
+
},
|
| 257 |
+
"peft": null,
|
| 258 |
+
"job": {
|
| 259 |
+
"target": null,
|
| 260 |
+
"image": "huggingface/lerobot-gpu:latest",
|
| 261 |
+
"timeout": "2d",
|
| 262 |
+
"detach": false,
|
| 263 |
+
"tags": []
|
| 264 |
+
},
|
| 265 |
+
"save_checkpoint_to_hub": false,
|
| 266 |
+
"sample_weighting": null,
|
| 267 |
+
"rename_map": {},
|
| 268 |
+
"checkpoint_path": null
|
| 269 |
+
}
|
Abs_6D/DiT_Layer/8/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/DiT_Layer/8/Baseline/checkpoints/030000/training_state/optimizer_param_groups.json
ADDED
|
@@ -0,0 +1,241 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
]
|
| 119 |
+
},
|
| 120 |
+
{
|
| 121 |
+
"weight_decay": 0.0,
|
| 122 |
+
"lr": 5.0659979290537954e-05,
|
| 123 |
+
"betas": [
|
| 124 |
+
0.9,
|
| 125 |
+
0.999
|
| 126 |
+
],
|
| 127 |
+
"eps": 1e-08,
|
| 128 |
+
"amsgrad": false,
|
| 129 |
+
"maximize": false,
|
| 130 |
+
"foreach": null,
|
| 131 |
+
"capturable": false,
|
| 132 |
+
"differentiable": false,
|
| 133 |
+
"fused": null,
|
| 134 |
+
"decoupled_weight_decay": true,
|
| 135 |
+
"initial_lr": 0.0001,
|
| 136 |
+
"params": [
|
| 137 |
+
99,
|
| 138 |
+
100,
|
| 139 |
+
101,
|
| 140 |
+
102,
|
| 141 |
+
103,
|
| 142 |
+
104,
|
| 143 |
+
105,
|
| 144 |
+
106,
|
| 145 |
+
107,
|
| 146 |
+
108,
|
| 147 |
+
109,
|
| 148 |
+
110,
|
| 149 |
+
111,
|
| 150 |
+
112,
|
| 151 |
+
113,
|
| 152 |
+
114,
|
| 153 |
+
115,
|
| 154 |
+
116,
|
| 155 |
+
117,
|
| 156 |
+
118,
|
| 157 |
+
119,
|
| 158 |
+
120,
|
| 159 |
+
121,
|
| 160 |
+
122,
|
| 161 |
+
123,
|
| 162 |
+
124,
|
| 163 |
+
125,
|
| 164 |
+
126,
|
| 165 |
+
127,
|
| 166 |
+
128,
|
| 167 |
+
129,
|
| 168 |
+
130,
|
| 169 |
+
131,
|
| 170 |
+
132,
|
| 171 |
+
133,
|
| 172 |
+
134,
|
| 173 |
+
135,
|
| 174 |
+
136,
|
| 175 |
+
137,
|
| 176 |
+
138,
|
| 177 |
+
139,
|
| 178 |
+
140,
|
| 179 |
+
141,
|
| 180 |
+
142,
|
| 181 |
+
143,
|
| 182 |
+
144,
|
| 183 |
+
145,
|
| 184 |
+
146,
|
| 185 |
+
147,
|
| 186 |
+
148,
|
| 187 |
+
149,
|
| 188 |
+
150,
|
| 189 |
+
151,
|
| 190 |
+
152,
|
| 191 |
+
153,
|
| 192 |
+
154,
|
| 193 |
+
155,
|
| 194 |
+
156,
|
| 195 |
+
157,
|
| 196 |
+
158,
|
| 197 |
+
159,
|
| 198 |
+
160,
|
| 199 |
+
161,
|
| 200 |
+
162,
|
| 201 |
+
163,
|
| 202 |
+
164,
|
| 203 |
+
165,
|
| 204 |
+
166,
|
| 205 |
+
167,
|
| 206 |
+
168,
|
| 207 |
+
169,
|
| 208 |
+
170,
|
| 209 |
+
171,
|
| 210 |
+
172,
|
| 211 |
+
173,
|
| 212 |
+
174,
|
| 213 |
+
175,
|
| 214 |
+
176,
|
| 215 |
+
177,
|
| 216 |
+
178,
|
| 217 |
+
179,
|
| 218 |
+
180,
|
| 219 |
+
181,
|
| 220 |
+
182,
|
| 221 |
+
183,
|
| 222 |
+
184,
|
| 223 |
+
185,
|
| 224 |
+
186,
|
| 225 |
+
187,
|
| 226 |
+
188,
|
| 227 |
+
189,
|
| 228 |
+
190,
|
| 229 |
+
191,
|
| 230 |
+
192,
|
| 231 |
+
193,
|
| 232 |
+
194,
|
| 233 |
+
195,
|
| 234 |
+
196,
|
| 235 |
+
197,
|
| 236 |
+
198,
|
| 237 |
+
199,
|
| 238 |
+
200
|
| 239 |
+
]
|
| 240 |
+
}
|
| 241 |
+
]
|
Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/training_state/training_step.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"step": 30000,
|
| 3 |
+
"num_processes": 1,
|
| 4 |
+
"batch_size": 64
|
| 5 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/debug.log
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
|
| 2 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Configure stats pid to 814701
|
| 3 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug.log
|
| 5 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug-internal.log
|
| 6 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:init():766] calling init triggers
|
| 7 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [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': [10]}}, '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/DiT_Layer/8/Baseline/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': None, '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, 'action_expert_num_layers': 8, 'rkd_enabled': False, 'rkd_vae_checkpoint': None, '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': 0.1, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 1.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline', 'job_name': 'INSIGHT_6D_DiT8_Baseline', 'resume': True, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 64, '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': '21sn30p5', '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/DiT_Layer/8/Baseline/checkpoints/030000', '_wandb': {}}
|
| 9 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:init():814] starting backend
|
| 10 |
+
2026-07-23 01:33:42,527 INFO MainThread:814701 [wandb_init.py:init():829] sending inform_init request
|
| 11 |
+
2026-07-23 01:33:42,987 INFO MainThread:814701 [wandb_init.py:init():834] backend started and connected
|
| 12 |
+
2026-07-23 01:33:42,989 INFO MainThread:814701 [wandb_init.py:init():904] updated telemetry
|
| 13 |
+
2026-07-23 01:33:42,993 INFO MainThread:814701 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-07-23 01:33:43,715 INFO MainThread:814701 [wandb_init.py:init():967] run resumed
|
| 15 |
+
2026-07-23 01:33:43,718 INFO MainThread:814701 [wandb_init.py:init():972] starting run threads in backend
|
| 16 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_console_start():2523] atexit reg
|
| 17 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2373] redirect: wrap_raw
|
| 18 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2442] Wrapping output streams.
|
| 19 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2465] Redirects installed.
|
| 20 |
+
2026-07-23 01:33:43,821 INFO MainThread:814701 [wandb_init.py:init():1010] run started, returning control to user process
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/files/output.log
ADDED
|
@@ -0,0 +1,231 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO 2026-07-22 02:08:01 db_utils.py:121 Logs will be synced with wandb.
|
| 2 |
+
INFO 2026-07-22 02:08:01 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/21sn30p5
|
| 3 |
+
INFO 2026-07-22 02:08:01 ot_train.py:298 Creating dataset
|
| 4 |
+
INFO 2026-07-22 02:08:03 ot_train.py:332 Creating policy
|
| 5 |
+
Fetching 27 files: 100%|ββββββββββ| 27/27 [00:00<00:00, 7327.95it/s]
|
| 6 |
+
`torch_dtype` is deprecated! Use `dtype` instead!
|
| 7 |
+
Loading weights: 100%|ββββββββββ| 695/695 [00:00<00:00, 824.80it/s]
|
| 8 |
+
[1mGR00TN17 LOAD REPORT[0m from: /home/ext_minje/.cache/huggingface/hub/models--nvidia--GR00T-N1.7-3B/snapshots/2fc962b973bccdd5d8ce4f67cc63b264d6886495
|
| 9 |
+
Key | Status | |
|
| 10 |
+
--------------------------------------------------------------------+------------+--+-
|
| 11 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.weight | UNEXPECTED | |
|
| 12 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.weight | UNEXPECTED | |
|
| 13 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.weight | UNEXPECTED | |
|
| 14 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.bias | UNEXPECTED | |
|
| 15 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.weight | UNEXPECTED | |
|
| 16 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.bias | UNEXPECTED | |
|
| 17 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.weight | UNEXPECTED | |
|
| 18 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.bias | UNEXPECTED | |
|
| 19 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.bias | UNEXPECTED | |
|
| 20 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.weight | UNEXPECTED | |
|
| 21 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.weight | UNEXPECTED | |
|
| 22 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.bias | UNEXPECTED | |
|
| 23 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.bias | UNEXPECTED | |
|
| 24 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.bias | UNEXPECTED | |
|
| 25 |
+
|
| 26 |
+
Notes:
|
| 27 |
+
- UNEXPECTED: can be ignored when loading from different task/architecture; not ok if you expect identical arch.
|
| 28 |
+
INFO 2026-07-22 02:08:09 ot_train.py:405 Creating optimizer and scheduler
|
| 29 |
+
INFO 2026-07-22 02:08:09 ot_train.py:437 Output dir: /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline
|
| 30 |
+
INFO 2026-07-22 02:08:09 ot_train.py:444 cfg.steps=60000 (60K)
|
| 31 |
+
INFO 2026-07-22 02:08:09 ot_train.py:445 dataset.num_frames=218367 (218K)
|
| 32 |
+
INFO 2026-07-22 02:08:09 ot_train.py:446 dataset.num_episodes=4930
|
| 33 |
+
INFO 2026-07-22 02:08:09 ot_train.py:449 Effective batch size: 64 x 1 = 64
|
| 34 |
+
INFO 2026-07-22 02:08:09 ot_train.py:450 num_learnable_params=808512640 (809M)
|
| 35 |
+
INFO 2026-07-22 02:08:09 ot_train.py:451 num_total_params=2333587585 (2B)
|
| 36 |
+
Training: 0%| | 0/60000 [00:00<?, ?step/s]INFO 2026-07-22 02:08:09 ot_train.py:602 Start offline training on a fixed dataset, with effective batch size: 64
|
| 37 |
+
Training: 0%| | 200/60000 [04:14<21:11:56, 1.28s/step]INFO 2026-07-22 02:12:23 ot_train.py:646 step:200 smpl:13K ep:289 epch:0.06 loss:0.998 grdn:1.416 lr:2.0e-05 updt_s:0.697 data_s:0.572 smp/s:50 mem_gb:24.86
|
| 38 |
+
Training: 1%| | 400/60000 [08:24<21:24:34, 1.29s/step]INFO 2026-07-22 02:16:34 ot_train.py:646 step:400 smpl:26K ep:578 epch:0.12 loss:0.315 grdn:1.801 lr:6.0e-05 updt_s:0.680 data_s:0.572 smp/s:51 mem_gb:24.88
|
| 39 |
+
Training: 1%| | 600/60000 [12:33<21:32:59, 1.31s/step]INFO 2026-07-22 02:20:42 ot_train.py:646 step:600 smpl:38K ep:867 epch:0.18 loss:0.229 grdn:1.092 lr:9.5e-05 updt_s:0.678 data_s:0.563 smp/s:52 mem_gb:24.88
|
| 40 |
+
Training: 1%|β | 800/60000 [16:41<20:49:44, 1.27s/step]INFO 2026-07-22 02:24:51 ot_train.py:646 step:800 smpl:51K ep:1K epch:0.23 loss:0.182 grdn:0.809 lr:1.0e-04 updt_s:0.676 data_s:0.567 smp/s:51 mem_gb:24.88
|
| 41 |
+
Training: 2%|β | 1000/60000 [20:51<20:28:48, 1.25s/step]INFO 2026-07-22 02:29:00 ot_train.py:646 step:1K smpl:64K ep:1K epch:0.29 loss:0.158 grdn:0.698 lr:1.0e-04 updt_s:0.680 data_s:0.564 smp/s:51 mem_gb:24.88
|
| 42 |
+
Training: 2%|β | 1200/60000 [24:59<20:04:58, 1.23s/step]INFO 2026-07-22 02:33:08 ot_train.py:646 step:1K smpl:77K ep:2K epch:0.35 loss:0.145 grdn:0.644 lr:1.0e-04 updt_s:0.676 data_s:0.564 smp/s:52 mem_gb:24.88
|
| 43 |
+
Training: 2%|β | 1400/60000 [29:08<20:01:36, 1.23s/step]INFO 2026-07-22 02:37:17 ot_train.py:646 step:1K smpl:90K ep:2K epch:0.41 loss:0.134 grdn:0.607 lr:1.0e-04 updt_s:0.677 data_s:0.566 smp/s:51 mem_gb:24.88
|
| 44 |
+
Training: 3%|β | 1600/60000 [33:16<19:54:58, 1.23s/step]INFO 2026-07-22 02:41:25 ot_train.py:646 step:2K smpl:102K ep:2K epch:0.47 loss:0.127 grdn:0.569 lr:1.0e-04 updt_s:0.675 data_s:0.563 smp/s:52 mem_gb:24.88
|
| 45 |
+
Training: 3%|β | 1800/60000 [37:25<21:26:17, 1.33s/step]INFO 2026-07-22 02:45:34 ot_train.py:646 step:2K smpl:115K ep:3K epch:0.53 loss:0.122 grdn:0.533 lr:1.0e-04 updt_s:0.680 data_s:0.565 smp/s:51 mem_gb:24.88
|
| 46 |
+
Training: 3%|β | 2000/60000 [41:42<20:16:58, 1.26s/step]INFO 2026-07-22 02:49:51 ot_train.py:646 step:2K smpl:128K ep:3K epch:0.59 loss:0.116 grdn:0.513 lr:1.0e-04 updt_s:0.689 data_s:0.593 smp/s:50 mem_gb:24.88
|
| 47 |
+
Training: 4%|β | 2200/60000 [46:03<20:39:54, 1.29s/step]INFO 2026-07-22 02:54:13 ot_train.py:646 step:2K smpl:141K ep:3K epch:0.64 loss:0.112 grdn:0.494 lr:1.0e-04 updt_s:0.717 data_s:0.590 smp/s:49 mem_gb:24.88
|
| 48 |
+
Training: 4%|β | 2400/60000 [50:19<20:15:54, 1.27s/step]INFO 2026-07-22 02:58:28 ot_train.py:646 step:2K smpl:154K ep:3K epch:0.70 loss:0.111 grdn:0.494 lr:1.0e-04 updt_s:0.700 data_s:0.578 smp/s:50 mem_gb:24.86
|
| 49 |
+
Training: 4%|β | 2600/60000 [54:27<20:09:02, 1.26s/step]INFO 2026-07-22 03:02:36 ot_train.py:646 step:3K smpl:166K ep:4K epch:0.76 loss:0.106 grdn:0.458 lr:1.0e-04 updt_s:0.677 data_s:0.561 smp/s:52 mem_gb:24.88
|
| 50 |
+
Training: 5%|β | 2800/60000 [58:44<25:30:01, 1.60s/step]INFO 2026-07-22 03:06:54 ot_train.py:646 step:3K smpl:179K ep:4K epch:0.82 loss:0.105 grdn:0.454 lr:1.0e-04 updt_s:0.720 data_s:0.567 smp/s:50 mem_gb:24.88
|
| 51 |
+
Training: 5%|β | 3000/60000 [1:02:57<19:20:08, 1.22s/step]INFO 2026-07-22 03:11:06 ot_train.py:646 step:3K smpl:192K ep:4K epch:0.88 loss:0.101 grdn:0.446 lr:1.0e-04 updt_s:0.700 data_s:0.560 smp/s:51 mem_gb:24.88
|
| 52 |
+
Training: 5%|β | 3200/60000 [1:07:10<19:46:37, 1.25s/step]INFO 2026-07-22 03:15:19 ot_train.py:646 step:3K smpl:205K ep:5K epch:0.94 loss:0.101 grdn:0.440 lr:1.0e-04 updt_s:0.704 data_s:0.562 smp/s:51 mem_gb:24.88
|
| 53 |
+
Training: 6%|β | 3400/60000 [1:11:19<20:13:27, 1.29s/step]INFO 2026-07-22 03:19:28 ot_train.py:646 step:3K smpl:218K ep:5K epch:1.00 loss:0.098 grdn:0.421 lr:9.9e-05 updt_s:0.677 data_s:0.565 smp/s:52 mem_gb:24.88
|
| 54 |
+
Training: 6%|β | 3600/60000 [1:16:26<19:35:52, 1.25s/step]INFO 2026-07-22 03:24:35 ot_train.py:646 step:4K smpl:230K ep:5K epch:1.06 loss:0.099 grdn:0.438 lr:9.9e-05 updt_s:0.932 data_s:0.603 smp/s:42 mem_gb:24.88
|
| 55 |
+
Training: 6%|β | 3800/60000 [1:20:32<18:55:33, 1.21s/step]INFO 2026-07-22 03:28:42 ot_train.py:646 step:4K smpl:243K ep:5K epch:1.11 loss:0.097 grdn:0.413 lr:9.9e-05 updt_s:0.675 data_s:0.556 smp/s:52 mem_gb:24.88
|
| 56 |
+
Training: 7%|β | 4000/60000 [1:24:58<19:26:31, 1.25s/step]INFO 2026-07-22 03:33:08 ot_train.py:646 step:4K smpl:256K ep:6K epch:1.17 loss:0.097 grdn:0.424 lr:9.9e-05 updt_s:0.752 data_s:0.576 smp/s:48 mem_gb:24.88
|
| 57 |
+
Training: 7%|β | 4200/60000 [1:29:09<18:58:07, 1.22s/step]INFO 2026-07-22 03:37:18 ot_train.py:646 step:4K smpl:269K ep:6K epch:1.23 loss:0.093 grdn:0.397 lr:9.9e-05 updt_s:0.685 data_s:0.567 smp/s:51 mem_gb:24.88
|
| 58 |
+
Training: 7%|β | 4400/60000 [1:33:18<18:58:27, 1.23s/step]INFO 2026-07-22 03:41:27 ot_train.py:646 step:4K smpl:282K ep:6K epch:1.29 loss:0.092 grdn:0.396 lr:9.9e-05 updt_s:0.678 data_s:0.566 smp/s:51 mem_gb:24.88
|
| 59 |
+
Training: 8%|β | 4600/60000 [1:37:29<20:44:42, 1.35s/step]INFO 2026-07-22 03:45:38 ot_train.py:646 step:5K smpl:294K ep:7K epch:1.35 loss:0.091 grdn:0.391 lr:9.9e-05 updt_s:0.682 data_s:0.572 smp/s:51 mem_gb:24.86
|
| 60 |
+
Training: 8%|β | 4800/60000 [1:41:38<19:24:01, 1.27s/step]INFO 2026-07-22 03:49:48 ot_train.py:646 step:5K smpl:307K ep:7K epch:1.41 loss:0.089 grdn:0.384 lr:9.9e-05 updt_s:0.679 data_s:0.567 smp/s:51 mem_gb:24.88
|
| 61 |
+
Training: 8%|β | 5000/60000 [1:45:48<18:49:33, 1.23s/step]INFO 2026-07-22 03:53:57 ot_train.py:646 step:5K smpl:320K ep:7K epch:1.47 loss:0.089 grdn:0.388 lr:9.9e-05 updt_s:0.682 data_s:0.565 smp/s:51 mem_gb:24.88
|
| 62 |
+
Training: 9%|β | 5200/60000 [1:49:58<19:21:44, 1.27s/step]INFO 2026-07-22 03:58:07 ot_train.py:646 step:5K smpl:333K ep:8K epch:1.52 loss:0.089 grdn:0.382 lr:9.9e-05 updt_s:0.679 data_s:0.570 smp/s:51 mem_gb:24.88
|
| 63 |
+
Training: 9%|β | 5400/60000 [1:54:09<18:45:50, 1.24s/step]INFO 2026-07-22 04:02:18 ot_train.py:646 step:5K smpl:346K ep:8K epch:1.58 loss:0.087 grdn:0.378 lr:9.8e-05 updt_s:0.686 data_s:0.568 smp/s:51 mem_gb:24.88
|
| 64 |
+
Training: 9%|β | 5600/60000 [1:58:18<18:35:47, 1.23s/step]INFO 2026-07-22 04:06:28 ot_train.py:646 step:6K smpl:358K ep:8K epch:1.64 loss:0.085 grdn:0.371 lr:9.8e-05 updt_s:0.679 data_s:0.566 smp/s:51 mem_gb:24.88
|
| 65 |
+
Training: 10%|β | 5800/60000 [2:02:27<18:19:49, 1.22s/step]INFO 2026-07-22 04:10:36 ot_train.py:646 step:6K smpl:371K ep:8K epch:1.70 loss:0.085 grdn:0.375 lr:9.8e-05 updt_s:0.683 data_s:0.560 smp/s:51 mem_gb:24.88
|
| 66 |
+
Training: 10%|β | 6000/60000 [2:06:37<18:32:42, 1.24s/step]INFO 2026-07-22 04:14:46 ot_train.py:646 step:6K smpl:384K ep:9K epch:1.76 loss:0.085 grdn:0.374 lr:9.8e-05 updt_s:0.680 data_s:0.567 smp/s:51 mem_gb:24.88
|
| 67 |
+
Training: 10%|β | 6200/60000 [2:10:45<18:24:04, 1.23s/step]INFO 2026-07-22 04:18:54 ot_train.py:646 step:6K smpl:397K ep:9K epch:1.82 loss:0.086 grdn:0.370 lr:9.8e-05 updt_s:0.670 data_s:0.569 smp/s:52 mem_gb:24.88
|
| 68 |
+
Training: 11%|β | 6400/60000 [2:14:59<18:33:11, 1.25s/step]INFO 2026-07-22 04:23:08 ot_train.py:646 step:6K smpl:410K ep:9K epch:1.88 loss:0.084 grdn:0.369 lr:9.8e-05 updt_s:0.699 data_s:0.570 smp/s:50 mem_gb:24.88
|
| 69 |
+
Training: 11%|β | 6600/60000 [2:19:19<18:30:20, 1.25s/step]INFO 2026-07-22 04:27:28 ot_train.py:646 step:7K smpl:422K ep:10K epch:1.93 loss:0.084 grdn:0.371 lr:9.8e-05 updt_s:0.702 data_s:0.597 smp/s:49 mem_gb:24.88
|
| 70 |
+
Training: 11%|ββ | 6800/60000 [2:23:38<19:19:14, 1.31s/step]INFO 2026-07-22 04:31:47 ot_train.py:646 step:7K smpl:435K ep:10K epch:1.99 loss:0.084 grdn:0.372 lr:9.7e-05 updt_s:0.693 data_s:0.602 smp/s:49 mem_gb:24.86
|
| 71 |
+
Training: 12%|ββ | 7000/60000 [2:27:54<18:28:29, 1.25s/step]INFO 2026-07-22 04:36:03 ot_train.py:646 step:7K smpl:448K ep:10K epch:2.05 loss:0.082 grdn:0.357 lr:9.7e-05 updt_s:0.695 data_s:0.584 smp/s:50 mem_gb:24.88
|
| 72 |
+
Training: 12%|ββ | 7200/60000 [2:32:29<21:37:48, 1.47s/step]INFO 2026-07-22 04:40:39 ot_train.py:646 step:7K smpl:461K ep:10K epch:2.11 loss:0.080 grdn:0.341 lr:9.7e-05 updt_s:0.720 data_s:0.655 smp/s:47 mem_gb:24.88
|
| 73 |
+
Training: 12%|ββ | 7400/60000 [2:37:38<23:24:55, 1.60s/step]INFO 2026-07-22 04:45:47 ot_train.py:646 step:7K smpl:474K ep:11K epch:2.17 loss:0.078 grdn:0.341 lr:9.7e-05 updt_s:0.806 data_s:0.737 smp/s:41 mem_gb:24.88
|
| 74 |
+
Training: 13%|ββ | 7600/60000 [2:42:52<21:01:02, 1.44s/step]INFO 2026-07-22 04:51:02 ot_train.py:646 step:8K smpl:486K ep:11K epch:2.23 loss:0.082 grdn:0.350 lr:9.7e-05 updt_s:0.818 data_s:0.751 smp/s:41 mem_gb:24.88
|
| 75 |
+
Training: 13%|ββ | 7800/60000 [2:48:06<22:22:47, 1.54s/step]INFO 2026-07-22 04:56:15 ot_train.py:646 step:8K smpl:499K ep:11K epch:2.29 loss:0.080 grdn:0.338 lr:9.6e-05 updt_s:0.760 data_s:0.805 smp/s:41 mem_gb:24.88
|
| 76 |
+
Training: 13%|ββ | 8000/60000 [2:53:55<21:53:35, 1.52s/step]INFO 2026-07-22 05:02:05 ot_train.py:646 step:8K smpl:512K ep:12K epch:2.34 loss:0.078 grdn:0.363 lr:9.6e-05 updt_s:0.746 data_s:1.002 smp/s:37 mem_gb:24.88
|
| 77 |
+
Training: 14%|ββ | 8200/60000 [3:00:13<26:36:53, 1.85s/step]INFO 2026-07-22 05:08:22 ot_train.py:646 step:8K smpl:525K ep:12K epch:2.40 loss:0.079 grdn:0.353 lr:9.6e-05 updt_s:0.781 data_s:1.105 smp/s:34 mem_gb:24.88
|
| 78 |
+
Training: 14%|ββ | 8400/60000 [3:06:20<22:59:19, 1.60s/step]INFO 2026-07-22 05:14:30 ot_train.py:646 step:8K smpl:538K ep:12K epch:2.46 loss:0.079 grdn:0.341 lr:9.6e-05 updt_s:0.754 data_s:1.082 smp/s:35 mem_gb:24.88
|
| 79 |
+
Training: 14%|ββ | 8600/60000 [3:12:30<29:57:33, 2.10s/step]INFO 2026-07-22 05:20:39 ot_train.py:646 step:9K smpl:550K ep:12K epch:2.52 loss:0.079 grdn:0.350 lr:9.6e-05 updt_s:0.768 data_s:1.077 smp/s:35 mem_gb:24.88
|
| 80 |
+
Training: 15%|ββ | 8800/60000 [3:17:12<17:47:07, 1.25s/step]INFO 2026-07-22 05:25:22 ot_train.py:646 step:9K smpl:563K ep:13K epch:2.58 loss:0.079 grdn:0.368 lr:9.5e-05 updt_s:0.693 data_s:0.718 smp/s:45 mem_gb:24.87
|
| 81 |
+
Training: 15%|ββ | 9000/60000 [3:21:23<18:02:52, 1.27s/step]INFO 2026-07-22 05:29:33 ot_train.py:646 step:9K smpl:576K ep:13K epch:2.64 loss:0.077 grdn:0.336 lr:9.5e-05 updt_s:0.714 data_s:0.540 smp/s:51 mem_gb:24.88
|
| 82 |
+
Training: 15%|ββ | 9200/60000 [3:26:27<22:53:21, 1.62s/step]INFO 2026-07-22 05:34:37 ot_train.py:646 step:9K smpl:589K ep:13K epch:2.70 loss:0.076 grdn:0.328 lr:9.5e-05 updt_s:0.773 data_s:0.745 smp/s:42 mem_gb:24.86
|
| 83 |
+
Training: 16%|ββ | 9400/60000 [3:31:33<21:26:44, 1.53s/step]INFO 2026-07-22 05:39:42 ot_train.py:646 step:9K smpl:602K ep:14K epch:2.75 loss:0.076 grdn:0.338 lr:9.5e-05 updt_s:0.781 data_s:0.744 smp/s:42 mem_gb:24.88
|
| 84 |
+
Training: 16%|ββ | 9600/60000 [3:36:38<21:53:41, 1.56s/step]INFO 2026-07-22 05:44:48 ot_train.py:646 step:10K smpl:614K ep:14K epch:2.81 loss:0.075 grdn:0.324 lr:9.4e-05 updt_s:0.779 data_s:0.748 smp/s:42 mem_gb:24.88
|
| 85 |
+
Training: 16%|ββ | 9800/60000 [3:42:22<22:22:02, 1.60s/step]INFO 2026-07-22 05:50:31 ot_train.py:646 step:10K smpl:627K ep:14K epch:2.87 loss:0.075 grdn:0.355 lr:9.4e-05 updt_s:0.756 data_s:0.960 smp/s:37 mem_gb:24.88
|
| 86 |
+
Training: 17%|ββ | 10000/60000 [3:47:46<32:39:13, 2.35s/step]INFO 2026-07-22 05:55:55 ot_train.py:646 step:10K smpl:640K ep:14K epch:2.93 loss:0.075 grdn:0.337 lr:9.4e-05 updt_s:0.773 data_s:0.847 smp/s:40 mem_gb:24.88
|
| 87 |
+
INFO 2026-07-22 05:55:55 ot_train.py:691 Checkpoint policy after step 10000
|
| 88 |
+
INFO 2026-07-22 05:56:28 in_yaml.py:1270 Saved config, action-space, prompt, and phase provenance in /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline and /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/checkpoints/010000
|
| 89 |
+
Training: 17%|ββ | 10200/60000 [3:55:05<20:20:07, 1.47s/step]INFO 2026-07-22 06:03:14 ot_train.py:646 step:10K smpl:653K ep:15K epch:2.99 loss:0.074 grdn:0.343 lr:9.4e-05 updt_s:0.756 data_s:1.275 smp/s:32 mem_gb:24.88
|
| 90 |
+
Training: 17%|ββ | 10400/60000 [4:00:45<24:23:20, 1.77s/step]INFO 2026-07-22 06:08:54 ot_train.py:646 step:10K smpl:666K ep:15K epch:3.05 loss:0.072 grdn:0.326 lr:9.3e-05 updt_s:0.773 data_s:0.926 smp/s:38 mem_gb:24.88
|
| 91 |
+
Training: 18%|ββ | 10600/60000 [4:06:14<21:35:25, 1.57s/step]INFO 2026-07-22 06:14:23 ot_train.py:646 step:11K smpl:678K ep:15K epch:3.11 loss:0.072 grdn:0.314 lr:9.3e-05 updt_s:0.780 data_s:0.865 smp/s:39 mem_gb:24.88
|
| 92 |
+
Training: 18%|ββ | 10800/60000 [4:11:29<23:55:18, 1.75s/step]INFO 2026-07-22 06:19:38 ot_train.py:646 step:11K smpl:691K ep:16K epch:3.17 loss:0.071 grdn:0.321 lr:9.3e-05 updt_s:0.818 data_s:0.754 smp/s:41 mem_gb:24.88
|
| 93 |
+
Training: 18%|ββ | 11000/60000 [4:16:49<23:03:08, 1.69s/step]INFO 2026-07-22 06:24:58 ot_train.py:646 step:11K smpl:704K ep:16K epch:3.22 loss:0.071 grdn:0.315 lr:9.3e-05 updt_s:0.799 data_s:0.800 smp/s:40 mem_gb:24.88
|
| 94 |
+
Training: 19%|ββ | 11200/60000 [4:22:43<22:54:57, 1.69s/step]INFO 2026-07-22 06:30:52 ot_train.py:646 step:11K smpl:717K ep:16K epch:3.28 loss:0.071 grdn:0.328 lr:9.2e-05 updt_s:0.798 data_s:0.969 smp/s:36 mem_gb:24.88
|
| 95 |
+
Training: 19%|ββ | 11400/60000 [4:28:27<23:45:49, 1.76s/step]INFO 2026-07-22 06:36:37 ot_train.py:646 step:11K smpl:730K ep:16K epch:3.34 loss:0.071 grdn:0.324 lr:9.2e-05 updt_s:0.761 data_s:0.962 smp/s:37 mem_gb:24.86
|
| 96 |
+
Training: 19%|ββ | 11600/60000 [4:34:42<21:32:20, 1.60s/step]INFO 2026-07-22 06:42:52 ot_train.py:646 step:12K smpl:742K ep:17K epch:3.40 loss:0.071 grdn:0.326 lr:9.2e-05 updt_s:0.872 data_s:1.002 smp/s:34 mem_gb:24.88
|
| 97 |
+
Training: 20%|ββ | 11800/60000 [4:40:19<20:46:08, 1.55s/step]INFO 2026-07-22 06:48:29 ot_train.py:646 step:12K smpl:755K ep:17K epch:3.46 loss:0.071 grdn:0.325 lr:9.2e-05 updt_s:0.826 data_s:0.857 smp/s:38 mem_gb:24.88
|
| 98 |
+
Training: 20%|ββ | 12000/60000 [4:45:52<22:37:29, 1.70s/step]INFO 2026-07-22 06:54:01 ot_train.py:646 step:12K smpl:768K ep:17K epch:3.52 loss:0.070 grdn:0.315 lr:9.1e-05 updt_s:0.818 data_s:0.843 smp/s:39 mem_gb:24.88
|
| 99 |
+
Training: 20%|ββ | 12200/60000 [4:51:12<21:25:54, 1.61s/step]INFO 2026-07-22 06:59:21 ot_train.py:646 step:12K smpl:781K ep:18K epch:3.58 loss:0.071 grdn:0.332 lr:9.1e-05 updt_s:0.836 data_s:0.761 smp/s:40 mem_gb:24.88
|
| 100 |
+
Training: 21%|ββ | 12400/60000 [4:57:15<28:09:50, 2.13s/step]INFO 2026-07-22 07:05:24 ot_train.py:646 step:12K smpl:794K ep:18K epch:3.63 loss:0.066 grdn:0.310 lr:9.1e-05 updt_s:0.796 data_s:1.016 smp/s:35 mem_gb:24.88
|
| 101 |
+
Training: 21%|ββ | 12600/60000 [5:03:48<21:34:55, 1.64s/step]INFO 2026-07-22 07:11:57 ot_train.py:646 step:13K smpl:806K ep:18K epch:3.69 loss:0.067 grdn:0.311 lr:9.0e-05 updt_s:0.796 data_s:1.168 smp/s:33 mem_gb:24.88
|
| 102 |
+
Training: 21%|βββ | 12800/60000 [5:09:01<17:55:21, 1.37s/step]INFO 2026-07-22 07:17:11 ot_train.py:646 step:13K smpl:819K ep:18K epch:3.75 loss:0.069 grdn:0.328 lr:9.0e-05 updt_s:0.806 data_s:0.761 smp/s:41 mem_gb:24.88
|
| 103 |
+
Training: 22%|βββ | 13000/60000 [5:13:28<16:58:26, 1.30s/step]INFO 2026-07-22 07:21:37 ot_train.py:646 step:13K smpl:832K ep:19K epch:3.81 loss:0.067 grdn:0.322 lr:9.0e-05 updt_s:0.721 data_s:0.611 smp/s:48 mem_gb:24.88
|
| 104 |
+
Training: 22%|βββ | 13200/60000 [5:18:07<20:37:49, 1.59s/step]INFO 2026-07-22 07:26:16 ot_train.py:646 step:13K smpl:845K ep:19K epch:3.87 loss:0.068 grdn:0.312 lr:8.9e-05 updt_s:0.742 data_s:0.650 smp/s:46 mem_gb:24.88
|
| 105 |
+
Training: 22%|βββ | 13400/60000 [5:23:01<19:10:58, 1.48s/step]INFO 2026-07-22 07:31:10 ot_train.py:646 step:13K smpl:858K ep:19K epch:3.93 loss:0.069 grdn:0.344 lr:8.9e-05 updt_s:0.755 data_s:0.715 smp/s:44 mem_gb:24.88
|
| 106 |
+
Training: 23%|βββ | 13600/60000 [5:27:33<16:20:55, 1.27s/step]INFO 2026-07-22 07:35:42 ot_train.py:646 step:14K smpl:870K ep:20K epch:3.99 loss:0.066 grdn:0.305 lr:8.9e-05 updt_s:0.721 data_s:0.636 smp/s:47 mem_gb:24.86
|
| 107 |
+
Training: 23%|βββ | 13800/60000 [5:31:53<17:09:20, 1.34s/step]INFO 2026-07-22 07:40:02 ot_train.py:646 step:14K smpl:883K ep:20K epch:4.04 loss:0.065 grdn:0.298 lr:8.8e-05 updt_s:0.697 data_s:0.602 smp/s:49 mem_gb:24.88
|
| 108 |
+
Training: 23%|βββ | 14000/60000 [5:36:12<16:42:34, 1.31s/step]INFO 2026-07-22 07:44:22 ot_train.py:646 step:14K smpl:896K ep:20K epch:4.10 loss:0.065 grdn:0.311 lr:8.8e-05 updt_s:0.726 data_s:0.571 smp/s:49 mem_gb:24.88
|
| 109 |
+
Training: 24%|βββ | 14200/60000 [5:40:29<16:00:21, 1.26s/step]INFO 2026-07-22 07:48:38 ot_train.py:646 step:14K smpl:909K ep:21K epch:4.16 loss:0.066 grdn:0.316 lr:8.8e-05 updt_s:0.727 data_s:0.553 smp/s:50 mem_gb:24.88
|
| 110 |
+
Training: 24%|βββ | 14400/60000 [5:44:41<15:52:49, 1.25s/step]INFO 2026-07-22 07:52:50 ot_train.py:646 step:14K smpl:922K ep:21K epch:4.22 loss:0.064 grdn:0.302 lr:8.7e-05 updt_s:0.711 data_s:0.548 smp/s:51 mem_gb:24.88
|
| 111 |
+
Training: 24%|βββ | 14600/60000 [5:48:55<15:46:51, 1.25s/step]INFO 2026-07-22 07:57:04 ot_train.py:646 step:15K smpl:934K ep:21K epch:4.28 loss:0.064 grdn:0.296 lr:8.7e-05 updt_s:0.718 data_s:0.551 smp/s:50 mem_gb:24.88
|
| 112 |
+
Training: 25%|βββ | 14800/60000 [5:53:10<15:56:25, 1.27s/step]INFO 2026-07-22 08:01:19 ot_train.py:646 step:15K smpl:947K ep:21K epch:4.34 loss:0.066 grdn:0.307 lr:8.7e-05 updt_s:0.715 data_s:0.558 smp/s:50 mem_gb:24.87
|
| 113 |
+
Training: 25%|βββ | 15000/60000 [5:57:24<15:41:01, 1.25s/step]INFO 2026-07-22 08:05:34 ot_train.py:646 step:15K smpl:960K ep:22K epch:4.40 loss:0.066 grdn:0.326 lr:8.6e-05 updt_s:0.719 data_s:0.553 smp/s:50 mem_gb:24.88
|
| 114 |
+
Training: 25%|βββ | 15200/60000 [6:01:40<15:33:04, 1.25s/step]INFO 2026-07-22 08:09:50 ot_train.py:646 step:15K smpl:973K ep:22K epch:4.45 loss:0.066 grdn:0.320 lr:8.6e-05 updt_s:0.715 data_s:0.565 smp/s:50 mem_gb:24.88
|
| 115 |
+
Training: 26%|βββ | 15400/60000 [6:05:54<16:21:02, 1.32s/step]INFO 2026-07-22 08:14:03 ot_train.py:646 step:15K smpl:986K ep:22K epch:4.51 loss:0.065 grdn:0.317 lr:8.5e-05 updt_s:0.703 data_s:0.561 smp/s:51 mem_gb:24.88
|
| 116 |
+
Training: 26%|βββ | 15600/60000 [6:10:02<15:55:40, 1.29s/step]INFO 2026-07-22 08:18:11 ot_train.py:646 step:16K smpl:998K ep:23K epch:4.57 loss:0.064 grdn:0.299 lr:8.5e-05 updt_s:0.670 data_s:0.571 smp/s:52 mem_gb:24.88
|
| 117 |
+
Training: 26%|βββ | 15800/60000 [6:14:24<19:33:31, 1.59s/step]INFO 2026-07-22 08:22:34 ot_train.py:646 step:16K smpl:1M ep:23K epch:4.63 loss:0.063 grdn:0.285 lr:8.5e-05 updt_s:0.747 data_s:0.564 smp/s:49 mem_gb:24.86
|
| 118 |
+
Training: 27%|βββ | 16000/60000 [6:18:46<16:07:30, 1.32s/step]INFO 2026-07-22 08:26:55 ot_train.py:646 step:16K smpl:1M ep:23K epch:4.69 loss:0.062 grdn:0.313 lr:8.4e-05 updt_s:0.751 data_s:0.555 smp/s:49 mem_gb:24.88
|
| 119 |
+
Training: 27%|βββ | 16200/60000 [6:23:20<18:23:48, 1.51s/step]INFO 2026-07-22 08:31:30 ot_train.py:646 step:16K smpl:1M ep:23K epch:4.75 loss:0.063 grdn:0.306 lr:8.4e-05 updt_s:0.817 data_s:0.555 smp/s:47 mem_gb:24.88
|
| 120 |
+
Training: 27%|βββ | 16400/60000 [6:27:55<17:01:37, 1.41s/step]INFO 2026-07-22 08:36:04 ot_train.py:646 step:16K smpl:1M ep:24K epch:4.81 loss:0.064 grdn:0.311 lr:8.4e-05 updt_s:0.803 data_s:0.566 smp/s:47 mem_gb:24.88
|
| 121 |
+
Training: 28%|βββ | 16600/60000 [6:32:30<18:03:20, 1.50s/step]INFO 2026-07-22 08:40:39 ot_train.py:646 step:17K smpl:1M ep:24K epch:4.87 loss:0.062 grdn:0.314 lr:8.3e-05 updt_s:0.802 data_s:0.573 smp/s:47 mem_gb:24.88
|
| 122 |
+
Training: 28%|βββ | 16800/60000 [6:36:52<17:04:46, 1.42s/step]INFO 2026-07-22 08:45:02 ot_train.py:646 step:17K smpl:1M ep:24K epch:4.92 loss:0.062 grdn:0.308 lr:8.3e-05 updt_s:0.747 data_s:0.564 smp/s:49 mem_gb:24.88
|
| 123 |
+
Training: 28%|βββ | 17000/60000 [6:41:23<17:41:22, 1.48s/step]INFO 2026-07-22 08:49:33 ot_train.py:646 step:17K smpl:1M ep:25K epch:4.98 loss:0.062 grdn:0.299 lr:8.2e-05 updt_s:0.791 data_s:0.561 smp/s:47 mem_gb:24.88
|
| 124 |
+
Training: 29%|βββ | 17200/60000 [6:45:43<16:50:33, 1.42s/step]INFO 2026-07-22 08:53:52 ot_train.py:646 step:17K smpl:1M ep:25K epch:5.04 loss:0.060 grdn:0.301 lr:8.2e-05 updt_s:0.728 data_s:0.568 smp/s:49 mem_gb:24.88
|
| 125 |
+
Training: 29%|βββ | 17400/60000 [6:50:02<16:07:28, 1.36s/step]INFO 2026-07-22 08:58:12 ot_train.py:646 step:17K smpl:1M ep:25K epch:5.10 loss:0.061 grdn:0.301 lr:8.2e-05 updt_s:0.734 data_s:0.563 smp/s:49 mem_gb:24.88
|
| 126 |
+
Training: 29%|βββ | 17600/60000 [6:54:37<16:17:11, 1.38s/step]INFO 2026-07-22 09:02:46 ot_train.py:646 step:18K smpl:1M ep:25K epch:5.16 loss:0.063 grdn:0.319 lr:8.1e-05 updt_s:0.812 data_s:0.559 smp/s:47 mem_gb:24.88
|
| 127 |
+
Training: 30%|βββ | 17800/60000 [6:59:07<17:39:57, 1.51s/step]INFO 2026-07-22 09:07:17 ot_train.py:646 step:18K smpl:1M ep:26K epch:5.22 loss:0.060 grdn:0.292 lr:8.1e-05 updt_s:0.777 data_s:0.576 smp/s:47 mem_gb:24.88
|
| 128 |
+
Training: 30%|βββ | 18000/60000 [7:03:37<14:33:00, 1.25s/step]INFO 2026-07-22 09:11:46 ot_train.py:646 step:18K smpl:1M ep:26K epch:5.28 loss:0.061 grdn:0.298 lr:8.0e-05 updt_s:0.780 data_s:0.567 smp/s:48 mem_gb:24.88
|
| 129 |
+
Training: 30%|βββ | 18200/60000 [7:08:05<16:49:00, 1.45s/step]INFO 2026-07-22 09:16:14 ot_train.py:646 step:18K smpl:1M ep:26K epch:5.33 loss:0.059 grdn:0.296 lr:8.0e-05 updt_s:0.782 data_s:0.556 smp/s:48 mem_gb:24.86
|
| 130 |
+
Training: 31%|βββ | 18400/60000 [7:12:40<15:12:30, 1.32s/step]INFO 2026-07-22 09:20:49 ot_train.py:646 step:18K smpl:1M ep:27K epch:5.39 loss:0.060 grdn:0.301 lr:7.9e-05 updt_s:0.823 data_s:0.550 smp/s:47 mem_gb:24.88
|
| 131 |
+
Training: 31%|βββ | 18600/60000 [7:17:20<15:49:10, 1.38s/step]INFO 2026-07-22 09:25:29 ot_train.py:646 step:19K smpl:1M ep:27K epch:5.45 loss:0.059 grdn:0.292 lr:7.9e-05 updt_s:0.838 data_s:0.563 smp/s:46 mem_gb:24.88
|
| 132 |
+
Training: 31%|ββββ | 18800/60000 [7:21:32<15:10:38, 1.33s/step]INFO 2026-07-22 09:29:42 ot_train.py:646 step:19K smpl:1M ep:27K epch:5.51 loss:0.059 grdn:0.287 lr:7.9e-05 updt_s:0.693 data_s:0.568 smp/s:51 mem_gb:24.88
|
| 133 |
+
Training: 32%|ββββ | 19000/60000 [7:26:03<16:07:17, 1.42s/step]INFO 2026-07-22 09:34:12 ot_train.py:646 step:19K smpl:1M ep:27K epch:5.57 loss:0.058 grdn:0.285 lr:7.8e-05 updt_s:0.789 data_s:0.562 smp/s:47 mem_gb:24.88
|
| 134 |
+
Training: 32%|ββββ | 19200/60000 [7:30:30<15:55:23, 1.40s/step]INFO 2026-07-22 09:38:39 ot_train.py:646 step:19K smpl:1M ep:28K epch:5.63 loss:0.060 grdn:0.306 lr:7.8e-05 updt_s:0.774 data_s:0.560 smp/s:48 mem_gb:24.88
|
| 135 |
+
Training: 32%|ββββ | 19400/60000 [7:34:54<14:40:34, 1.30s/step]INFO 2026-07-22 09:43:03 ot_train.py:646 step:19K smpl:1M ep:28K epch:5.69 loss:0.059 grdn:0.295 lr:7.7e-05 updt_s:0.766 data_s:0.554 smp/s:48 mem_gb:24.88
|
| 136 |
+
Training: 33%|ββββ | 19600/60000 [7:39:34<16:53:21, 1.50s/step]INFO 2026-07-22 09:47:44 ot_train.py:646 step:20K smpl:1M ep:28K epch:5.74 loss:0.059 grdn:0.297 lr:7.7e-05 updt_s:0.836 data_s:0.564 smp/s:46 mem_gb:24.88
|
| 137 |
+
Training: 33%|ββββ | 19800/60000 [7:44:08<13:35:51, 1.22s/step]INFO 2026-07-22 09:52:18 ot_train.py:646 step:20K smpl:1M ep:29K epch:5.80 loss:0.058 grdn:0.296 lr:7.6e-05 updt_s:0.812 data_s:0.556 smp/s:47 mem_gb:24.88
|
| 138 |
+
Training: 33%|ββββ | 20000/60000 [7:48:31<14:58:08, 1.35s/step]INFO 2026-07-22 09:56:41 ot_train.py:646 step:20K smpl:1M ep:29K epch:5.86 loss:0.058 grdn:0.292 lr:7.6e-05 updt_s:0.764 data_s:0.552 smp/s:49 mem_gb:24.88
|
| 139 |
+
Training: 34%|ββββ | 20200/60000 [7:53:02<16:05:27, 1.46s/step]INFO 2026-07-22 10:01:11 ot_train.py:646 step:20K smpl:1M ep:29K epch:5.92 loss:0.058 grdn:0.289 lr:7.6e-05 updt_s:0.784 data_s:0.565 smp/s:47 mem_gb:24.88
|
| 140 |
+
Training: 34%|ββββ | 20400/60000 [7:57:26<16:04:48, 1.46s/step]INFO 2026-07-22 10:05:35 ot_train.py:646 step:20K smpl:1M ep:29K epch:5.98 loss:0.055 grdn:0.291 lr:7.5e-05 updt_s:0.754 data_s:0.565 smp/s:49 mem_gb:24.86
|
| 141 |
+
Training: 34%|ββββ | 20600/60000 [8:01:44<13:45:53, 1.26s/step]INFO 2026-07-22 10:09:53 ot_train.py:646 step:21K smpl:1M ep:30K epch:6.04 loss:0.056 grdn:0.288 lr:7.5e-05 updt_s:0.739 data_s:0.551 smp/s:50 mem_gb:24.88
|
| 142 |
+
Training: 35%|ββββ | 20800/60000 [8:06:00<14:23:45, 1.32s/step]INFO 2026-07-22 10:14:09 ot_train.py:646 step:21K smpl:1M ep:30K epch:6.10 loss:0.056 grdn:0.296 lr:7.4e-05 updt_s:0.730 data_s:0.548 smp/s:50 mem_gb:24.88
|
| 143 |
+
Training: 35%|ββββ | 21000/60000 [8:10:16<13:17:13, 1.23s/step]INFO 2026-07-22 10:18:26 ot_train.py:646 step:21K smpl:1M ep:30K epch:6.15 loss:0.057 grdn:0.297 lr:7.4e-05 updt_s:0.732 data_s:0.549 smp/s:50 mem_gb:24.88
|
| 144 |
+
Training: 35%|ββββ | 21200/60000 [8:14:44<15:53:47, 1.47s/step]INFO 2026-07-22 10:22:53 ot_train.py:646 step:21K smpl:1M ep:31K epch:6.21 loss:0.054 grdn:0.289 lr:7.3e-05 updt_s:0.763 data_s:0.574 smp/s:48 mem_gb:24.88
|
| 145 |
+
Training: 36%|ββββ | 21400/60000 [8:19:52<15:26:18, 1.44s/step]INFO 2026-07-22 10:28:02 ot_train.py:646 step:21K smpl:1M ep:31K epch:6.27 loss:0.055 grdn:0.294 lr:7.3e-05 updt_s:0.849 data_s:0.692 smp/s:42 mem_gb:24.88
|
| 146 |
+
Training: 36%|ββββ | 21600/60000 [8:25:01<16:01:00, 1.50s/step]INFO 2026-07-22 10:33:11 ot_train.py:646 step:22K smpl:1M ep:31K epch:6.33 loss:0.056 grdn:0.297 lr:7.2e-05 updt_s:0.872 data_s:0.672 smp/s:41 mem_gb:24.88
|
| 147 |
+
Training: 36%|ββββ | 21800/60000 [8:30:05<16:20:32, 1.54s/step]INFO 2026-07-22 10:38:14 ot_train.py:646 step:22K smpl:1M ep:31K epch:6.39 loss:0.056 grdn:0.285 lr:7.2e-05 updt_s:0.890 data_s:0.627 smp/s:42 mem_gb:24.88
|
| 148 |
+
Training: 37%|ββββ | 22000/60000 [8:35:11<16:22:35, 1.55s/step]INFO 2026-07-22 10:43:20 ot_train.py:646 step:22K smpl:1M ep:32K epch:6.45 loss:0.055 grdn:0.283 lr:7.1e-05 updt_s:0.888 data_s:0.640 smp/s:42 mem_gb:24.88
|
| 149 |
+
Training: 37%|ββββ | 22200/60000 [8:40:14<16:03:38, 1.53s/step]INFO 2026-07-22 10:48:24 ot_train.py:646 step:22K smpl:1M ep:32K epch:6.51 loss:0.055 grdn:0.282 lr:7.1e-05 updt_s:0.873 data_s:0.643 smp/s:42 mem_gb:24.88
|
| 150 |
+
Training: 37%|ββββ | 22400/60000 [8:45:21<16:01:42, 1.53s/step]INFO 2026-07-22 10:53:30 ot_train.py:646 step:22K smpl:1M ep:32K epch:6.57 loss:0.056 grdn:0.292 lr:7.0e-05 updt_s:0.886 data_s:0.645 smp/s:42 mem_gb:24.88
|
| 151 |
+
Training: 38%|ββββ | 22600/60000 [8:50:34<16:11:00, 1.56s/step]INFO 2026-07-22 10:58:44 ot_train.py:646 step:23K smpl:1M ep:33K epch:6.62 loss:0.056 grdn:0.296 lr:7.0e-05 updt_s:0.901 data_s:0.666 smp/s:41 mem_gb:24.86
|
| 152 |
+
Training: 38%|ββββ | 22800/60000 [8:55:43<16:13:58, 1.57s/step]INFO 2026-07-22 11:03:53 ot_train.py:646 step:23K smpl:1M ep:33K epch:6.68 loss:0.054 grdn:0.282 lr:6.9e-05 updt_s:0.891 data_s:0.653 smp/s:41 mem_gb:24.88
|
| 153 |
+
Training: 38%|ββββ | 23000/60000 [9:00:53<16:19:17, 1.59s/step]INFO 2026-07-22 11:09:02 ot_train.py:646 step:23K smpl:1M ep:33K epch:6.74 loss:0.053 grdn:0.286 lr:6.9e-05 updt_s:0.907 data_s:0.640 smp/s:41 mem_gb:24.88
|
| 154 |
+
Training: 39%|ββββ | 23200/60000 [9:05:42<14:14:54, 1.39s/step]INFO 2026-07-22 11:13:52 ot_train.py:646 step:23K smpl:1M ep:34K epch:6.80 loss:0.053 grdn:0.285 lr:6.8e-05 updt_s:0.796 data_s:0.649 smp/s:44 mem_gb:24.88
|
| 155 |
+
Training: 39%|ββββ | 23400/60000 [9:10:20<14:01:20, 1.38s/step]INFO 2026-07-22 11:18:29 ot_train.py:646 step:23K smpl:1M ep:34K epch:6.86 loss:0.053 grdn:0.290 lr:6.8e-05 updt_s:0.746 data_s:0.643 smp/s:46 mem_gb:24.88
|
| 156 |
+
Training: 39%|ββββ | 23600/60000 [9:14:59<13:50:06, 1.37s/step]INFO 2026-07-22 11:23:09 ot_train.py:646 step:24K smpl:2M ep:34K epch:6.92 loss:0.053 grdn:0.279 lr:6.7e-05 updt_s:0.741 data_s:0.654 smp/s:46 mem_gb:24.88
|
| 157 |
+
Training: 40%|ββοΏ½οΏ½οΏ½β | 23800/60000 [9:19:47<14:06:22, 1.40s/step]INFO 2026-07-22 11:27:57 ot_train.py:646 step:24K smpl:2M ep:34K epch:6.98 loss:0.053 grdn:0.283 lr:6.7e-05 updt_s:0.764 data_s:0.675 smp/s:44 mem_gb:24.88
|
| 158 |
+
Training: 40%|ββββ | 24000/60000 [9:24:37<13:43:49, 1.37s/step]INFO 2026-07-22 11:32:46 ot_train.py:646 step:24K smpl:2M ep:35K epch:7.03 loss:0.051 grdn:0.290 lr:6.6e-05 updt_s:0.766 data_s:0.679 smp/s:44 mem_gb:24.88
|
| 159 |
+
Training: 40%|ββββ | 24200/60000 [9:29:13<14:00:11, 1.41s/step]INFO 2026-07-22 11:37:22 ot_train.py:646 step:24K smpl:2M ep:35K epch:7.09 loss:0.052 grdn:0.290 lr:6.6e-05 updt_s:0.725 data_s:0.652 smp/s:46 mem_gb:24.88
|
| 160 |
+
Training: 41%|ββββ | 24400/60000 [9:33:56<13:54:38, 1.41s/step]INFO 2026-07-22 11:42:05 ot_train.py:646 step:24K smpl:2M ep:35K epch:7.15 loss:0.052 grdn:0.285 lr:6.5e-05 updt_s:0.747 data_s:0.667 smp/s:45 mem_gb:24.88
|
| 161 |
+
Training: 41%|ββββ | 24600/60000 [9:38:40<14:29:37, 1.47s/step]INFO 2026-07-22 11:46:49 ot_train.py:646 step:25K smpl:2M ep:36K epch:7.21 loss:0.053 grdn:0.290 lr:6.5e-05 updt_s:0.755 data_s:0.665 smp/s:45 mem_gb:24.88
|
| 162 |
+
Training: 41%|βββββ | 24800/60000 [9:43:22<14:48:20, 1.51s/step]INFO 2026-07-22 11:51:31 ot_train.py:646 step:25K smpl:2M ep:36K epch:7.27 loss:0.053 grdn:0.303 lr:6.4e-05 updt_s:0.750 data_s:0.658 smp/s:45 mem_gb:24.88
|
| 163 |
+
Training: 42%|βββββ | 25000/60000 [9:48:20<15:33:02, 1.60s/step]INFO 2026-07-22 11:56:29 ot_train.py:646 step:25K smpl:2M ep:36K epch:7.33 loss:0.050 grdn:0.284 lr:6.4e-05 updt_s:0.795 data_s:0.691 smp/s:43 mem_gb:24.86
|
| 164 |
+
Training: 42%|βββββ | 25200/60000 [9:53:18<13:45:52, 1.42s/step]INFO 2026-07-22 12:01:27 ot_train.py:646 step:25K smpl:2M ep:36K epch:7.39 loss:0.050 grdn:0.281 lr:6.3e-05 updt_s:0.818 data_s:0.671 smp/s:43 mem_gb:24.88
|
| 165 |
+
Training: 42%|βββββ | 25400/60000 [9:58:00<13:57:37, 1.45s/step]INFO 2026-07-22 12:06:10 ot_train.py:646 step:25K smpl:2M ep:37K epch:7.44 loss:0.050 grdn:0.287 lr:6.3e-05 updt_s:0.759 data_s:0.652 smp/s:45 mem_gb:24.88
|
| 166 |
+
Training: 43%|βββββ | 25600/60000 [10:02:39<13:39:59, 1.43s/step]INFO 2026-07-22 12:10:48 ot_train.py:646 step:26K smpl:2M ep:37K epch:7.50 loss:0.051 grdn:0.287 lr:6.2e-05 updt_s:0.734 data_s:0.658 smp/s:46 mem_gb:24.88
|
| 167 |
+
Training: 43%|βββββ | 25800/60000 [10:07:20<15:54:11, 1.67s/step]INFO 2026-07-22 12:15:30 ot_train.py:646 step:26K smpl:2M ep:37K epch:7.56 loss:0.050 grdn:0.289 lr:6.2e-05 updt_s:0.746 data_s:0.660 smp/s:46 mem_gb:24.88
|
| 168 |
+
Training: 43%|βββββ | 26000/60000 [10:12:01<12:40:47, 1.34s/step]INFO 2026-07-22 12:20:11 ot_train.py:646 step:26K smpl:2M ep:38K epch:7.62 loss:0.049 grdn:0.290 lr:6.1e-05 updt_s:0.735 data_s:0.666 smp/s:46 mem_gb:24.88
|
| 169 |
+
Training: 44%|βββββ | 26200/60000 [10:16:53<14:08:28, 1.51s/step]INFO 2026-07-22 12:25:02 ot_train.py:646 step:26K smpl:2M ep:38K epch:7.68 loss:0.050 grdn:0.289 lr:6.1e-05 updt_s:0.765 data_s:0.692 smp/s:44 mem_gb:24.88
|
| 170 |
+
Training: 44%|βββββ | 26400/60000 [10:21:40<13:41:57, 1.47s/step]INFO 2026-07-22 12:29:49 ot_train.py:646 step:26K smpl:2M ep:38K epch:7.74 loss:0.051 grdn:0.299 lr:6.0e-05 updt_s:0.746 data_s:0.689 smp/s:45 mem_gb:24.88
|
| 171 |
+
Training: 44%|βββββ | 26600/60000 [10:26:24<13:13:58, 1.43s/step]INFO 2026-07-22 12:34:34 ot_train.py:646 step:27K smpl:2M ep:38K epch:7.80 loss:0.050 grdn:0.286 lr:6.0e-05 updt_s:0.751 data_s:0.668 smp/s:45 mem_gb:24.88
|
| 172 |
+
Training: 45%|βββββ | 26800/60000 [10:31:02<12:57:31, 1.41s/step]INFO 2026-07-22 12:39:11 ot_train.py:646 step:27K smpl:2M ep:39K epch:7.85 loss:0.049 grdn:0.283 lr:5.9e-05 updt_s:0.731 data_s:0.657 smp/s:46 mem_gb:24.88
|
| 173 |
+
Training: 45%|βββββ | 27000/60000 [10:35:43<12:52:32, 1.40s/step]INFO 2026-07-22 12:43:53 ot_train.py:646 step:27K smpl:2M ep:39K epch:7.91 loss:0.050 grdn:0.293 lr:5.9e-05 updt_s:0.740 data_s:0.665 smp/s:46 mem_gb:24.88
|
| 174 |
+
Training: 45%|βββββ | 27200/60000 [10:40:24<12:18:21, 1.35s/step]INFO 2026-07-22 12:48:33 ot_train.py:646 step:27K smpl:2M ep:39K epch:7.97 loss:0.048 grdn:0.287 lr:5.8e-05 updt_s:0.732 data_s:0.670 smp/s:46 mem_gb:24.86
|
| 175 |
+
Training: 46%|βββββ | 27400/60000 [10:45:08<12:35:56, 1.39s/step]INFO 2026-07-22 12:53:17 ot_train.py:646 step:27K smpl:2M ep:40K epch:8.03 loss:0.047 grdn:0.285 lr:5.8e-05 updt_s:0.753 data_s:0.664 smp/s:45 mem_gb:24.88
|
| 176 |
+
Training: 46%|βββββ | 27600/60000 [10:49:30<11:01:36, 1.23s/step]INFO 2026-07-22 12:57:39 ot_train.py:646 step:28K smpl:2M ep:40K epch:8.09 loss:0.047 grdn:0.292 lr:5.7e-05 updt_s:0.698 data_s:0.610 smp/s:49 mem_gb:24.88
|
| 177 |
+
Training: 46%|βββββ | 27800/60000 [10:53:35<10:47:52, 1.21s/step]INFO 2026-07-22 13:01:44 ot_train.py:646 step:28K smpl:2M ep:40K epch:8.15 loss:0.046 grdn:0.296 lr:5.7e-05 updt_s:0.661 data_s:0.564 smp/s:52 mem_gb:24.88
|
| 178 |
+
Training: 47%|βββββ | 28000/60000 [10:57:49<10:56:27, 1.23s/step]INFO 2026-07-22 13:05:58 ot_train.py:646 step:28K smpl:2M ep:40K epch:8.21 loss:0.047 grdn:0.297 lr:5.6e-05 updt_s:0.674 data_s:0.592 smp/s:51 mem_gb:24.88
|
| 179 |
+
Training: 47%|βββββ | 28200/60000 [11:01:56<10:39:32, 1.21s/step]INFO 2026-07-22 13:10:06 ot_train.py:646 step:28K smpl:2M ep:41K epch:8.26 loss:0.046 grdn:0.287 lr:5.6e-05 updt_s:0.660 data_s:0.578 smp/s:52 mem_gb:24.88
|
| 180 |
+
Training: 47%|βββββ | 28400/60000 [11:06:30<14:06:15, 1.61s/step]INFO 2026-07-22 13:14:39 ot_train.py:646 step:28K smpl:2M ep:41K epch:8.32 loss:0.045 grdn:0.287 lr:5.5e-05 updt_s:0.719 data_s:0.648 smp/s:47 mem_gb:24.88
|
| 181 |
+
Training: 48%|βββββ | 28600/60000 [11:11:02<11:42:32, 1.34s/step]INFO 2026-07-22 13:19:11 ot_train.py:646 step:29K smpl:2M ep:41K epch:8.38 loss:0.047 grdn:0.302 lr:5.5e-05 updt_s:0.723 data_s:0.634 smp/s:47 mem_gb:24.88
|
| 182 |
+
Training: 48%|βββββ | 28800/60000 [11:15:36<12:04:26, 1.39s/step]INFO 2026-07-22 13:23:45 ot_train.py:646 step:29K smpl:2M ep:42K epch:8.44 loss:0.046 grdn:0.290 lr:5.4e-05 updt_s:0.716 data_s:0.652 smp/s:47 mem_gb:24.88
|
| 183 |
+
Training: 48%|βββββ | 29000/60000 [11:20:02<11:33:37, 1.34s/step]INFO 2026-07-22 13:28:11 ot_train.py:646 step:29K smpl:2M ep:42K epch:8.50 loss:0.046 grdn:0.296 lr:5.4e-05 updt_s:0.699 data_s:0.629 smp/s:48 mem_gb:24.88
|
| 184 |
+
Training: 49%|βββββ | 29200/60000 [11:24:35<11:20:18, 1.33s/step]INFO 2026-07-22 13:32:44 ot_train.py:646 step:29K smpl:2M ep:42K epch:8.56 loss:0.046 grdn:0.290 lr:5.3e-05 updt_s:0.717 data_s:0.646 smp/s:47 mem_gb:24.88
|
| 185 |
+
Training: 49%|βββββ | 29400/60000 [11:29:18<11:33:05, 1.36s/step]INFO 2026-07-22 13:37:28 ot_train.py:646 step:29K smpl:2M ep:42K epch:8.62 loss:0.045 grdn:0.301 lr:5.3e-05 updt_s:0.757 data_s:0.661 smp/s:45 mem_gb:24.86
|
| 186 |
+
Training: 49%|βββββ | 29600/60000 [11:34:02<12:46:10, 1.51s/step]INFO 2026-07-22 13:42:11 ot_train.py:646 step:30K smpl:2M ep:43K epch:8.68 loss:0.045 grdn:0.305 lr:5.2e-05 updt_s:0.739 data_s:0.675 smp/s:45 mem_gb:24.88
|
| 187 |
+
Training: 50%|βββββ | 29800/60000 [11:38:57<11:47:19, 1.41s/step]INFO 2026-07-22 13:47:06 ot_train.py:646 step:30K smpl:2M ep:43K epch:8.73 loss:0.044 grdn:0.282 lr:5.1e-05 updt_s:0.764 data_s:0.712 smp/s:43 mem_gb:24.88
|
| 188 |
+
Training: 50%|βββββ | 30000/60000 [11:43:44<11:26:03, 1.37s/step]INFO 2026-07-22 13:51:53 ot_train.py:646 step:30K smpl:2M ep:43K epch:8.79 loss:0.043 grdn:0.286 lr:5.1e-05 updt_s:0.744 data_s:0.690 smp/s:45 mem_gb:24.88
|
| 189 |
+
INFO 2026-07-22 13:51:53 ot_train.py:691 Checkpoint policy after step 30000
|
| 190 |
+
INFO 2026-07-22 13:52:15 in_yaml.py:1270 Saved config, action-space, prompt, and phase provenance in /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline and /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000
|
| 191 |
+
Training: 50%|βββββ | 30200/60000 [11:48:57<11:36:46, 1.40s/step]INFO 2026-07-22 13:57:06 ot_train.py:646 step:30K smpl:2M ep:44K epch:8.85 loss:0.044 grdn:0.285 lr:5.0e-05 updt_s:0.768 data_s:0.686 smp/s:44 mem_gb:24.88
|
| 192 |
+
Training: 51%|βββββ | 30400/60000 [11:53:43<11:31:59, 1.40s/step]INFO 2026-07-22 14:01:52 ot_train.py:646 step:30K smpl:2M ep:44K epch:8.91 loss:0.043 grdn:0.280 lr:5.0e-05 updt_s:0.745 data_s:0.683 smp/s:45 mem_gb:24.88
|
| 193 |
+
Training: 51%|βββββ | 30600/60000 [11:58:29<11:47:56, 1.44s/step]INFO 2026-07-22 14:06:39 ot_train.py:646 step:31K smpl:2M ep:44K epch:8.97 loss:0.044 grdn:0.296 lr:4.9e-05 updt_s:0.756 data_s:0.675 smp/s:45 mem_gb:24.88
|
| 194 |
+
Training: 51%|ββββββ | 30800/60000 [12:03:46<12:09:01, 1.50s/step]INFO 2026-07-22 14:11:55 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.03 loss:0.044 grdn:0.283 lr:4.9e-05 updt_s:0.766 data_s:0.816 smp/s:40 mem_gb:24.88
|
| 195 |
+
Training: 52%|ββββββ | 31000/60000 [12:09:07<13:21:13, 1.66s/step]INFO 2026-07-22 14:17:17 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.09 loss:0.044 grdn:0.294 lr:4.8e-05 updt_s:0.741 data_s:0.865 smp/s:40 mem_gb:24.88
|
| 196 |
+
Training: 52%|ββββββ | 31200/60000 [12:14:49<18:49:57, 2.35s/step]INFO 2026-07-22 14:22:58 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.14 loss:0.044 grdn:0.290 lr:4.8e-05 updt_s:0.722 data_s:0.983 smp/s:38 mem_gb:24.88
|
| 197 |
+
Training: 52%|ββββββ | 31400/60000 [12:24:03<12:42:22, 1.60s/step]INFO 2026-07-22 14:32:13 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.20 loss:0.044 grdn:0.303 lr:4.7e-05 updt_s:0.788 data_s:1.984 smp/s:23 mem_gb:24.88
|
| 198 |
+
Training: 53%|ββββββ | 31600/60000 [12:29:18<14:47:59, 1.88s/step]INFO 2026-07-22 14:37:28 ot_train.py:646 step:32K smpl:2M ep:46K epch:9.26 loss:0.044 grdn:0.292 lr:4.7e-05 updt_s:0.785 data_s:0.789 smp/s:41 mem_gb:24.86
|
| 199 |
+
Training: 53%|ββββββ | 31800/60000 [12:35:03<12:30:52, 1.60s/step]INFO 2026-07-22 14:43:12 ot_train.py:646 step:32K smpl:2M ep:46K epch:9.32 loss:0.041 grdn:0.287 lr:4.6e-05 updt_s:0.788 data_s:0.931 smp/s:37 mem_gb:24.88
|
| 200 |
+
Training: 53%|ββββββ | 32000/60000 [12:41:10<14:22:55, 1.85s/step]INFO 2026-07-22 14:49:19 ot_train.py:646 step:32K smpl:2M ep:46K epch:9.38 loss:0.040 grdn:0.290 lr:4.6e-05 updt_s:0.761 data_s:1.073 smp/s:35 mem_gb:24.88
|
| 201 |
+
Training: 54%|ββββββ | 32200/60000 [12:46:37<11:54:17, 1.54s/step]INFO 2026-07-22 14:54:46 ot_train.py:646 step:32K smpl:2M ep:47K epch:9.44 loss:0.041 grdn:0.295 lr:4.5e-05 updt_s:0.768 data_s:0.866 smp/s:39 mem_gb:24.88
|
| 202 |
+
Training: 54%|ββββββ | 32400/60000 [12:52:10<12:02:31, 1.57s/step]INFO 2026-07-22 15:00:20 ot_train.py:646 step:32K smpl:2M ep:47K epch:9.50 loss:0.041 grdn:0.308 lr:4.5e-05 updt_s:0.769 data_s:0.897 smp/s:38 mem_gb:24.88
|
| 203 |
+
Training: 54%|ββββββ | 32600/60000 [12:57:34<14:03:50, 1.85s/step]INFO 2026-07-22 15:05:44 ot_train.py:646 step:33K smpl:2M ep:47K epch:9.55 loss:0.040 grdn:0.300 lr:4.4e-05 updt_s:0.762 data_s:0.856 smp/s:40 mem_gb:24.88
|
| 204 |
+
Training: 55%|ββββββ | 32800/60000 [13:03:03<11:52:48, 1.57s/step]INFO 2026-07-22 15:11:12 ot_train.py:646 step:33K smpl:2M ep:47K epch:9.61 loss:0.040 grdn:0.303 lr:4.4e-05 updt_s:0.770 data_s:0.872 smp/s:39 mem_gb:24.88
|
| 205 |
+
Training: 55%|ββββββ | 33000/60000 [13:08:38<12:59:53, 1.73s/step]INFO 2026-07-22 15:16:48 ot_train.py:646 step:33K smpl:2M ep:48K epch:9.67 loss:0.042 grdn:0.295 lr:4.3e-05 updt_s:0.743 data_s:0.931 smp/s:38 mem_gb:24.88
|
| 206 |
+
Training: 55%|ββββββ | 33200/60000 [13:14:16<11:33:29, 1.55s/step]INFO 2026-07-22 15:22:25 ot_train.py:646 step:33K smpl:2M ep:48K epch:9.73 loss:0.041 grdn:0.294 lr:4.3e-05 updt_s:0.725 data_s:0.961 smp/s:38 mem_gb:24.88
|
| 207 |
+
Training: 56%|ββββββ | 33400/60000 [13:19:56<11:32:29, 1.56s/step]INFO 2026-07-22 15:28:05 ot_train.py:646 step:33K smpl:2M ep:48K epch:9.79 loss:0.040 grdn:0.294 lr:4.2e-05 updt_s:0.728 data_s:0.972 smp/s:38 mem_gb:24.88
|
| 208 |
+
Training: 56%|ββββββ | 33600/60000 [13:25:08<12:02:53, 1.64s/step]INFO 2026-07-22 15:33:18 ot_train.py:646 step:34K smpl:2M ep:49K epch:9.85 loss:0.040 grdn:0.294 lr:4.1e-05 updt_s:0.743 data_s:0.817 smp/s:41 mem_gb:24.88
|
| 209 |
+
Training: 56%|ββββββ | 33800/60000 [13:30:26<10:54:32, 1.50s/step]INFO 2026-07-22 15:38:35 ot_train.py:646 step:34K smpl:2M ep:49K epch:9.91 loss:0.040 grdn:0.297 lr:4.1e-05 updt_s:0.773 data_s:0.812 smp/s:40 mem_gb:24.88
|
| 210 |
+
Training: 57%|ββββββ | 34000/60000 [13:36:00<15:31:44, 2.15s/step]INFO 2026-07-22 15:44:09 ot_train.py:646 step:34K smpl:2M ep:49K epch:9.96 loss:0.040 grdn:0.298 lr:4.0e-05 updt_s:0.754 data_s:0.913 smp/s:38 mem_gb:24.86
|
| 211 |
+
Training: 57%|ββββββ | 34200/60000 [13:41:35<10:47:58, 1.51s/step]INFO 2026-07-22 15:49:44 ot_train.py:646 step:34K smpl:2M ep:49K epch:10.02 loss:0.038 grdn:0.294 lr:4.0e-05 updt_s:0.727 data_s:0.947 smp/s:38 mem_gb:24.88
|
| 212 |
+
Training: 57%|ββββββ | 34400/60000 [13:46:39<12:07:09, 1.70s/step]INFO 2026-07-22 15:54:48 ot_train.py:646 step:34K smpl:2M ep:50K epch:10.08 loss:0.037 grdn:0.298 lr:3.9e-05 updt_s:0.741 data_s:0.777 smp/s:42 mem_gb:24.88
|
| 213 |
+
Training: 58%|ββββββ | 34600/60000 [13:51:56<10:08:27, 1.44s/step]INFO 2026-07-22 16:00:06 ot_train.py:646 step:35K smpl:2M ep:50K epch:10.14 loss:0.038 grdn:0.300 lr:3.9e-05 updt_s:0.731 data_s:0.855 smp/s:40 mem_gb:24.88
|
| 214 |
+
Training: 58%|ββββββ | 34800/60000 [13:56:58<11:00:25, 1.57s/step]INFO 2026-07-22 16:05:08 ot_train.py:646 step:35K smpl:2M ep:50K epch:10.20 loss:0.038 grdn:0.302 lr:3.8e-05 updt_s:0.761 data_s:0.750 smp/s:42 mem_gb:24.88
|
| 215 |
+
Training: 58%|ββββββ | 35000/60000 [14:02:10<11:37:58, 1.68s/step]INFO 2026-07-22 16:10:19 ot_train.py:646 step:35K smpl:2M ep:51K epch:10.26 loss:0.038 grdn:0.303 lr:3.8e-05 updt_s:0.763 data_s:0.791 smp/s:41 mem_gb:24.88
|
| 216 |
+
Training: 59%|ββββββ | 35200/60000 [14:07:16<10:32:44, 1.53s/step]INFO 2026-07-22 16:15:25 ot_train.py:646 step:35K smpl:2M ep:51K epch:10.32 loss:0.038 grdn:0.304 lr:3.7e-05 updt_s:0.764 data_s:0.766 smp/s:42 mem_gb:24.88
|
| 217 |
+
Training: 59%|ββββββ | 35400/60000 [14:12:20<8:39:25, 1.27s/step]INFO 2026-07-22 16:20:30 ot_train.py:646 step:35K smpl:2M ep:51K epch:10.38 loss:0.038 grdn:0.295 lr:3.7e-05 updt_s:0.746 data_s:0.774 smp/s:42 mem_gb:24.88
|
| 218 |
+
Training: 59%|ββββββ | 35600/60000 [14:16:27<8:00:40, 1.18s/step]INFO 2026-07-22 16:24:36 ot_train.py:646 step:36K smpl:2M ep:51K epch:10.43 loss:0.036 grdn:0.297 lr:3.6e-05 updt_s:0.652 data_s:0.579 smp/s:52 mem_gb:24.88
|
| 219 |
+
Training: 60%|ββββββ | 35800/60000 [14:20:52<8:17:04, 1.23s/step]INFO 2026-07-22 16:29:01 ot_train.py:646 step:36K smpl:2M ep:52K epch:10.49 loss:0.037 grdn:0.301 lr:3.6e-05 updt_s:0.669 data_s:0.656 smp/s:48 mem_gb:24.88
|
| 220 |
+
Training: 60%|ββββββ | 36000/60000 [14:26:01<8:48:44, 1.32s/step]INFO 2026-07-22 16:34:10 ot_train.py:646 step:36K smpl:2M ep:52K epch:10.55 loss:0.036 grdn:0.296 lr:3.5e-05 updt_s:0.670 data_s:0.874 smp/s:41 mem_gb:24.88
|
| 221 |
+
Training: 60%|ββββββ | 36200/60000 [14:30:20<8:21:08, 1.26s/step]INFO 2026-07-22 16:38:30 ot_train.py:646 step:36K smpl:2M ep:52K epch:10.61 loss:0.037 grdn:0.312 lr:3.5e-05 updt_s:0.661 data_s:0.634 smp/s:49 mem_gb:24.86
|
| 222 |
+
Training: 61%|ββββββ | 36400/60000 [14:34:27<7:58:48, 1.22s/step]INFO 2026-07-22 16:42:37 ot_train.py:646 step:36K smpl:2M ep:53K epch:10.67 loss:0.036 grdn:0.302 lr:3.4e-05 updt_s:0.660 data_s:0.574 smp/s:52 mem_gb:24.88
|
| 223 |
+
Training: 61%|ββββββ | 36600/60000 [14:38:34<7:51:28, 1.21s/step]INFO 2026-07-22 16:46:43 ot_train.py:646 step:37K smpl:2M ep:53K epch:10.73 loss:0.035 grdn:0.305 lr:3.4e-05 updt_s:0.658 data_s:0.573 smp/s:52 mem_gb:24.88
|
| 224 |
+
Training: 61%|βββββββ | 36800/60000 [14:42:40<7:52:05, 1.22s/step]INFO 2026-07-22 16:50:49 ot_train.py:646 step:37K smpl:2M ep:53K epch:10.79 loss:0.034 grdn:0.295 lr:3.3e-05 updt_s:0.653 data_s:0.576 smp/s:52 mem_gb:24.88
|
| 225 |
+
Training: 62%|βββββββ | 37000/60000 [14:46:44<7:41:43, 1.20s/step]INFO 2026-07-22 16:54:54 ot_train.py:646 step:37K smpl:2M ep:53K epch:10.84 loss:0.035 grdn:0.293 lr:3.3e-05 updt_s:0.648 data_s:0.572 smp/s:52 mem_gb:24.88
|
| 226 |
+
Training: 62%|βββββββ | 37200/60000 [14:50:49<7:51:36, 1.24s/step]INFO 2026-07-22 16:58:58 ot_train.py:646 step:37K smpl:2M ep:54K epch:10.90 loss:0.035 grdn:0.288 lr:3.2e-05 updt_s:0.657 data_s:0.565 smp/s:52 mem_gb:24.88
|
| 227 |
+
Training: 62%|βββββββ | 37400/60000 [14:54:55<7:25:48, 1.18s/step]INFO 2026-07-22 17:03:04 ot_train.py:646 step:37K smpl:2M ep:54K epch:10.96 loss:0.035 grdn:0.299 lr:3.2e-05 updt_s:0.658 data_s:0.570 smp/s:52 mem_gb:24.87
|
| 228 |
+
Training: 63%|βββββββ | 37600/60000 [14:58:59<7:36:30, 1.22s/step]INFO 2026-07-22 17:07:09 ot_train.py:646 step:38K smpl:2M ep:54K epch:11.02 loss:0.034 grdn:0.292 lr:3.1e-05 updt_s:0.658 data_s:0.563 smp/s:52 mem_gb:24.88
|
| 229 |
+
Training: 63%|βββββββ | 37800/60000 [15:03:02<7:17:53, 1.18s/step]INFO 2026-07-22 17:11:12 ot_train.py:646 step:38K smpl:2M ep:55K epch:11.08 loss:0.035 grdn:0.308 lr:3.1e-05 updt_s:0.651 data_s:0.563 smp/s:53 mem_gb:24.88
|
| 230 |
+
Training: 63%|βββββββ | 38000/60000 [15:07:03<7:27:23, 1.22s/step]INFO 2026-07-22 17:15:13 ot_train.py:646 step:38K smpl:2M ep:55K epch:11.14 loss:0.036 grdn:0.300 lr:3.0e-05 updt_s:0.648 data_s:0.555 smp/s:53 mem_gb:24.88
|
| 231 |
+
Training: 63%|βββββββ | 38009/60000 [15:07:15<7:40:05, 1.26s/step]
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/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/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug-core.log
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-07-22T02:08:00.119464938Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpnpx_8n8e/port-126141.txt","pid":126141,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 2 |
+
{"time":"2026-07-22T02:08:00.120016994Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":126141}
|
| 3 |
+
{"time":"2026-07-22T02:08:00.120017139Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-126141-130041-1189763398/socket","Net":"unix"}}
|
| 4 |
+
{"time":"2026-07-22T02:08:00.169736727Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp_i85ijjb/port-126152.txt","pid":126152,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 5 |
+
{"time":"2026-07-22T02:08:00.170930965Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":126152}
|
| 6 |
+
{"time":"2026-07-22T02:08:00.170919431Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-126152-130082-3741661681/socket","Net":"unix"}}
|
| 7 |
+
{"time":"2026-07-22T02:08:00.303798114Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 8 |
+
{"time":"2026-07-22T02:08:00.310000814Z","level":"INFO","msg":"handleInformInit: received","streamId":"rd8tz18f","id":"1(@)"}
|
| 9 |
+
{"time":"2026-07-22T02:08:00.355199917Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 10 |
+
{"time":"2026-07-22T02:08:00.361348251Z","level":"INFO","msg":"handleInformInit: received","streamId":"21sn30p5","id":"1(@)"}
|
| 11 |
+
{"time":"2026-07-22T02:08:00.773273199Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"rd8tz18f","id":"1(@)"}
|
| 12 |
+
{"time":"2026-07-22T02:08:00.817155993Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"21sn30p5","id":"1(@)"}
|
| 13 |
+
{"time":"2026-07-22T02:08:06.409472162Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"wnw3iz6fbp6m"}
|
| 14 |
+
{"time":"2026-07-22T02:08:06.509993434Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"l0fljn6r9zin"}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug-internal.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug.log
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
|
| 2 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_setup.py:_flush():81] Configure stats pid to 126152
|
| 3 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug.log
|
| 5 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260722_020800-21sn30p5/logs/debug-internal.log
|
| 6 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [wandb_init.py:init():766] calling init triggers
|
| 7 |
+
2026-07-22 02:08:00,138 INFO MainThread:126152 [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': {}, '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_steps': None, '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, 'action_expert_num_layers': 8, 'rkd_enabled': False, 'rkd_vae_checkpoint': None, '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': 0.1, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 1.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline', 'job_name': 'INSIGHT_6D_DiT8_Baseline', 'resume': False, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 64, '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-22 02:08:00,138 INFO MainThread:126152 [wandb_init.py:init():814] starting backend
|
| 10 |
+
2026-07-22 02:08:00,355 INFO MainThread:126152 [wandb_init.py:init():829] sending inform_init request
|
| 11 |
+
2026-07-22 02:08:00,818 INFO MainThread:126152 [wandb_init.py:init():834] backend started and connected
|
| 12 |
+
2026-07-22 02:08:00,820 INFO MainThread:126152 [wandb_init.py:init():904] updated telemetry
|
| 13 |
+
2026-07-22 02:08:00,824 INFO MainThread:126152 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-07-22 02:08:01,373 INFO MainThread:126152 [wandb_init.py:init():972] starting run threads in backend
|
| 15 |
+
2026-07-22 02:08:01,428 INFO MainThread:126152 [wandb_run.py:_console_start():2523] atexit reg
|
| 16 |
+
2026-07-22 02:08:01,429 INFO MainThread:126152 [wandb_run.py:_redirect():2373] redirect: wrap_raw
|
| 17 |
+
2026-07-22 02:08:01,429 INFO MainThread:126152 [wandb_run.py:_redirect():2442] Wrapping output streams.
|
| 18 |
+
2026-07-22 02:08:01,429 INFO MainThread:126152 [wandb_run.py:_redirect():2465] Redirects installed.
|
| 19 |
+
2026-07-22 02:08:01,431 INFO MainThread:126152 [wandb_init.py:init():1010] run started, returning control to user process
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/output.log
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO 2026-07-23 00:58:51 db_utils.py:121 Logs will be synced with wandb.
|
| 2 |
+
INFO 2026-07-23 00:58:51 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/21sn30p5
|
| 3 |
+
INFO 2026-07-23 00:58:51 ot_train.py:298 Creating dataset
|
| 4 |
+
INFO 2026-07-23 00:58:53 ot_train.py:332 Creating policy
|
| 5 |
+
INFO 2026-07-23 00:58:53 ng_groot.py:193 The Groot policy wraps NVIDIA's GR00T n1.7 model. Loading pretrained model from: /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model
|
| 6 |
+
INFO 2026-07-23 00:58:53 ng_groot.py:227 Detected fine-tuned LeRobot checkpoint, loading with state dict...
|
| 7 |
+
Fetching 27 files: 100%|ββββββββββ| 27/27 [00:00<00:00, 13208.10it/s]
|
| 8 |
+
`torch_dtype` is deprecated! Use `dtype` instead!
|
| 9 |
+
Loading weights: 100%|ββββββββββ| 695/695 [00:03<00:00, 198.66it/s]
|
| 10 |
+
[1mGR00TN17 LOAD REPORT[0m from: /home/ext_minje/.cache/huggingface/hub/models--nvidia--GR00T-N1.7-3B/snapshots/2fc962b973bccdd5d8ce4f67cc63b264d6886495
|
| 11 |
+
Key | Status | |
|
| 12 |
+
--------------------------------------------------------------------+------------+--+-
|
| 13 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.weight | UNEXPECTED | |
|
| 14 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.bias | UNEXPECTED | |
|
| 15 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.weight | UNEXPECTED | |
|
| 16 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.weight | UNEXPECTED | |
|
| 17 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.bias | UNEXPECTED | |
|
| 18 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.weight | UNEXPECTED | |
|
| 19 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.weight | UNEXPECTED | |
|
| 20 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.bias | UNEXPECTED | |
|
| 21 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.bias | UNEXPECTED | |
|
| 22 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.bias | UNEXPECTED | |
|
| 23 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.bias | UNEXPECTED | |
|
| 24 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.bias | UNEXPECTED | |
|
| 25 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.weight | UNEXPECTED | |
|
| 26 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.weight | UNEXPECTED | |
|
| 27 |
+
|
| 28 |
+
Notes:
|
| 29 |
+
- UNEXPECTED: can be ignored when loading from different task/architecture; not ok if you expect identical arch.
|
| 30 |
+
Loading weights from local directory
|
| 31 |
+
INFO 2026-07-23 00:59:17 ot_train.py:405 Creating optimizer and scheduler
|
| 32 |
+
INFO 2026-07-23 00:59:21 ot_train.py:437 Output dir: /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline
|
| 33 |
+
INFO 2026-07-23 00:59:21 ot_train.py:444 cfg.steps=60000 (60K)
|
| 34 |
+
INFO 2026-07-23 00:59:21 ot_train.py:445 dataset.num_frames=218367 (218K)
|
| 35 |
+
INFO 2026-07-23 00:59:21 ot_train.py:446 dataset.num_episodes=4930
|
| 36 |
+
INFO 2026-07-23 00:59:21 ot_train.py:449 Effective batch size: 64 x 1 = 64
|
| 37 |
+
INFO 2026-07-23 00:59:21 ot_train.py:450 num_learnable_params=808512640 (809M)
|
| 38 |
+
INFO 2026-07-23 00:59:21 ot_train.py:451 num_total_params=2333587585 (2B)
|
| 39 |
+
INFO 2026-07-23 00:59:21 ot_train.py:492 Resuming data order at epoch 13, sample 42176
|
| 40 |
+
Training: 0%| | 0/30000 [00:00<?, ?step/s]INFO 2026-07-23 00:59:21 ot_train.py:602 Start offline training on a fixed dataset, with effective batch size: 64
|
| 41 |
+
Training: 1%| | 200/30000 [04:30<10:53:15, 1.32s/step]INFO 2026-07-23 01:03:51 ot_train.py:646 step:30K smpl:2M ep:44K epch:8.85 loss:0.043 grdn:0.291 lr:5.0e-05 updt_s:0.739 data_s:0.612 smp/s:47 mem_gb:24.87
|
| 42 |
+
Training: 1%|β | 400/30000 [08:54<11:12:58, 1.36s/step]INFO 2026-07-23 01:08:16 ot_train.py:646 step:30K smpl:2M ep:44K epch:8.91 loss:0.043 grdn:0.295 lr:5.0e-05 updt_s:0.742 data_s:0.580 smp/s:48 mem_gb:24.87
|
| 43 |
+
Training: 2%|β | 600/30000 [13:18<11:03:42, 1.35s/step]INFO 2026-07-23 01:12:39 ot_train.py:646 step:31K smpl:2M ep:44K epch:8.97 loss:0.044 grdn:0.296 lr:4.9e-05 updt_s:0.735 data_s:0.582 smp/s:49 mem_gb:24.87
|
| 44 |
+
Training: 3%|β | 800/30000 [18:02<11:23:00, 1.40s/step]INFO 2026-07-23 01:17:23 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.03 loss:0.044 grdn:0.285 lr:4.9e-05 updt_s:0.778 data_s:0.639 smp/s:45 mem_gb:24.87
|
| 45 |
+
Training: 3%|β | 1000/30000 [23:25<12:54:18, 1.60s/step]INFO 2026-07-23 01:22:47 ot_train.py:646 step:31K smpl:2M ep:45K epch:9.09 loss:0.042 grdn:0.283 lr:4.8e-05 updt_s:0.816 data_s:0.800 smp/s:40 mem_gb:24.87
|
| 46 |
+
Training: 4%|β | 1145/30000 [27:30<19:41:45, 2.46s/step]Traceback (most recent call last):
|
| 47 |
+
File "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py", line 1453, in <module>
|
| 48 |
+
main()
|
| 49 |
+
File "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py", line 1449, in main
|
| 50 |
+
train_module.main()
|
| 51 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/scripts/lerobot_train.py", line 813, in main
|
| 52 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/configs/parser.py", line 320, in wrapper_inner
|
| 53 |
+
response = fn(cfg, *args, **kwargs)
|
| 54 |
+
^^^^^^^^^^^^^^^^^^^^^^^^
|
| 55 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/scripts/lerobot_train.py", line 612, in train
|
| 56 |
+
if sample_weighter is not None:
|
| 57 |
+
^^^^^^^^^^^^^^^^^^^
|
| 58 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/processor/pipeline.py", line 299, in __call__
|
| 59 |
+
transformed_transition = self._forward(transition)
|
| 60 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 61 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/processor/pipeline.py", line 316, in _forward
|
| 62 |
+
transition = processor_step(transition)
|
| 63 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 64 |
+
File "/home/ext_minje/clvla/lerobot/src/lerobot/policies/groot/processor_groot.py", line 1850, in __call__
|
| 65 |
+
video = np.stack(cams, axis=2) # (B, T, V, H, W, C)
|
| 66 |
+
^^^^^^^^^^^^^^^^^^^^^^
|
| 67 |
+
File "/home/ext_minje/miniconda3/envs/lerobot060_groot/lib/python3.12/site-packages/numpy/_core/shape_base.py", line 467, in stack
|
| 68 |
+
return _nx.concatenate(expanded_arrays, axis=axis, out=out,
|
| 69 |
+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
| 70 |
+
KeyboardInterrupt
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/wandb-metadata.json
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"os": "Linux-6.8.0-55-generic-x86_64-with-glibc2.39",
|
| 3 |
+
"python": "CPython 3.12.13",
|
| 4 |
+
"startedAt": "2026-07-23T00:58:50.451520Z",
|
| 5 |
+
"args": [
|
| 6 |
+
"--config_path=/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model/train_config.json",
|
| 7 |
+
"--resume=true"
|
| 8 |
+
],
|
| 9 |
+
"program": "/home/ext_minje/clvla/benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
|
| 10 |
+
"codePath": "benchmarks/INSIGHT/my_scripts/lerobot_train_yaml.py",
|
| 11 |
+
"codePathLocal": "my_scripts/lerobot_train_yaml.py",
|
| 12 |
+
"git": {
|
| 13 |
+
"remote": "git@github.com:paragon7060/clvla.git",
|
| 14 |
+
"commit": "a329728c2c9524d79b966c4a7a82ee56a4c2abf7"
|
| 15 |
+
},
|
| 16 |
+
"root": "/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline",
|
| 17 |
+
"host": "ext_minje-box",
|
| 18 |
+
"executable": "/home/ext_minje/miniconda3/envs/lerobot060_groot/bin/python",
|
| 19 |
+
"cpu_count": 112,
|
| 20 |
+
"cpu_count_logical": 224,
|
| 21 |
+
"gpu": "NVIDIA H200",
|
| 22 |
+
"gpu_count": 2,
|
| 23 |
+
"disk": {
|
| 24 |
+
"/": {
|
| 25 |
+
"total": "1888425144320",
|
| 26 |
+
"used": "1135434260480"
|
| 27 |
+
}
|
| 28 |
+
},
|
| 29 |
+
"memory": {
|
| 30 |
+
"total": "2164176568320"
|
| 31 |
+
},
|
| 32 |
+
"gpu_nvidia": [
|
| 33 |
+
{
|
| 34 |
+
"name": "NVIDIA H200",
|
| 35 |
+
"memoryTotal": "150754820096",
|
| 36 |
+
"cudaCores": 16896,
|
| 37 |
+
"architecture": "Hopper",
|
| 38 |
+
"uuid": "GPU-86be6e5b-fb48-fa1c-9bd7-83d336cb2991"
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"name": "NVIDIA H200",
|
| 42 |
+
"memoryTotal": "150754820096",
|
| 43 |
+
"cudaCores": 16896,
|
| 44 |
+
"architecture": "Hopper",
|
| 45 |
+
"uuid": "GPU-24eb1ba2-7313-ff93-1435-748768ebdbdc"
|
| 46 |
+
}
|
| 47 |
+
],
|
| 48 |
+
"cudaVersion": "12.8",
|
| 49 |
+
"writerId": "00spweoz26l63hzys0f5f482plpkbhoe"
|
| 50 |
+
}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/files/wandb-summary.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"_wandb":{"runtime":55870},"train/steps":37800,"_timestamp":1.7847402722368646e+09,"train/epochs":11.078597040761654,"train/lr":3.083421446984113e-05,"train/update_s":0.6514109047697275,"train/dataloading_s":0.5634355993702047,"train/samples_per_s":52.68155259277773,"_step":37800,"train/grad_norm":0.30793055780231954,"train/gpu_mem_gb":24.87638006925583,"train/samples":2419200,"train/episodes":54617.48341095496,"train/loss":0.03277819976210594,"_runtime":55870}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug-core.log
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-07-23T00:58:50.442526721Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp2ior1pmp/port-454.txt","pid":454,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 2 |
+
{"time":"2026-07-23T00:58:50.443023801Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-454-1268-2688650548/socket","Net":"unix"}}
|
| 3 |
+
{"time":"2026-07-23T00:58:50.44313088Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":454}
|
| 4 |
+
{"time":"2026-07-23T00:58:50.483257536Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpubgbbv96/port-430.txt","pid":430,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 5 |
+
{"time":"2026-07-23T00:58:50.483763212Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":430}
|
| 6 |
+
{"time":"2026-07-23T00:58:50.483763342Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-430-1294-1550743546/socket","Net":"unix"}}
|
| 7 |
+
{"time":"2026-07-23T00:58:50.487589039Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp9cf47msf/port-438.txt","pid":438,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 8 |
+
{"time":"2026-07-23T00:58:50.48889134Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":438}
|
| 9 |
+
{"time":"2026-07-23T00:58:50.488890648Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-438-1300-2209329374/socket","Net":"unix"}}
|
| 10 |
+
{"time":"2026-07-23T00:58:50.53986185Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpdlqfoxj9/port-446.txt","pid":446,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
|
| 11 |
+
{"time":"2026-07-23T00:58:50.541192813Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":446}
|
| 12 |
+
{"time":"2026-07-23T00:58:50.541189694Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-446-1338-2725561867/socket","Net":"unix"}}
|
| 13 |
+
{"time":"2026-07-23T00:58:50.628929549Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 14 |
+
{"time":"2026-07-23T00:58:50.635401913Z","level":"INFO","msg":"handleInformInit: received","streamId":"pgr3eari","id":"1(@)"}
|
| 15 |
+
{"time":"2026-07-23T00:58:50.667984858Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 16 |
+
{"time":"2026-07-23T00:58:50.673941153Z","level":"INFO","msg":"handleInformInit: received","streamId":"21sn30p5","id":"1(@)"}
|
| 17 |
+
{"time":"2026-07-23T00:58:50.67404224Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 18 |
+
{"time":"2026-07-23T00:58:50.680727357Z","level":"INFO","msg":"handleInformInit: received","streamId":"rd8tz18f","id":"1(@)"}
|
| 19 |
+
{"time":"2026-07-23T00:58:50.725811626Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
|
| 20 |
+
{"time":"2026-07-23T00:58:50.731237223Z","level":"INFO","msg":"handleInformInit: received","streamId":"svym8k0w","id":"1(@)"}
|
| 21 |
+
{"time":"2026-07-23T00:58:51.09301534Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"pgr3eari","id":"1(@)"}
|
| 22 |
+
{"time":"2026-07-23T00:58:51.120058787Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"rd8tz18f","id":"1(@)"}
|
| 23 |
+
{"time":"2026-07-23T00:58:51.131344474Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"21sn30p5","id":"1(@)"}
|
| 24 |
+
{"time":"2026-07-23T00:58:51.185176771Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"svym8k0w","id":"1(@)"}
|
| 25 |
+
{"time":"2026-07-23T00:58:56.889415247Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"4rih9cljf6h7"}
|
| 26 |
+
{"time":"2026-07-23T00:58:56.894112928Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"w89tjw1p8ewg"}
|
| 27 |
+
{"time":"2026-07-23T00:58:56.928667081Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"2jvc25k8t0qw"}
|
| 28 |
+
{"time":"2026-07-23T00:58:57.00426725Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"jtm5vnlo5mvy"}
|
| 29 |
+
{"time":"2026-07-23T01:26:52.217924187Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
|
| 30 |
+
{"time":"2026-07-23T01:26:52.218131434Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
|
| 31 |
+
{"time":"2026-07-23T01:26:52.218172554Z","level":"INFO","msg":"server is shutting down"}
|
| 32 |
+
{"time":"2026-07-23T01:26:52.218155392Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
|
| 33 |
+
{"time":"2026-07-23T01:26:52.218211496Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
|
| 34 |
+
{"time":"2026-07-23T01:26:52.218261727Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-454-1268-2688650548/socket","Net":"unix"}}
|
| 35 |
+
{"time":"2026-07-23T01:26:52.226525402Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
|
| 36 |
+
{"time":"2026-07-23T01:26:52.226598688Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
|
| 37 |
+
{"time":"2026-07-23T01:26:52.226616087Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
|
| 38 |
+
{"time":"2026-07-23T01:26:52.227419595Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
|
| 39 |
+
{"time":"2026-07-23T01:26:52.226627243Z","level":"INFO","msg":"server is shutting down"}
|
| 40 |
+
{"time":"2026-07-23T01:26:52.227486909Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-438-1300-2209329374/socket","Net":"unix"}}
|
| 41 |
+
{"time":"2026-07-23T01:26:52.239974032Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
|
| 42 |
+
{"time":"2026-07-23T01:26:52.24005161Z","level":"INFO","msg":"server is shutting down"}
|
| 43 |
+
{"time":"2026-07-23T01:26:52.24004295Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
|
| 44 |
+
{"time":"2026-07-23T01:26:52.240094238Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
|
| 45 |
+
{"time":"2026-07-23T01:26:52.240192908Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
|
| 46 |
+
{"time":"2026-07-23T01:26:52.240180686Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-446-1338-2725561867/socket","Net":"unix"}}
|
| 47 |
+
{"time":"2026-07-23T01:26:52.365092389Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
|
| 48 |
+
{"time":"2026-07-23T01:26:52.365205336Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
|
| 49 |
+
{"time":"2026-07-23T01:26:52.365271549Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
|
| 50 |
+
{"time":"2026-07-23T01:26:52.365280497Z","level":"INFO","msg":"server is shutting down"}
|
| 51 |
+
{"time":"2026-07-23T01:26:52.365368244Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-430-1294-1550743546/socket","Net":"unix"}}
|
| 52 |
+
{"time":"2026-07-23T01:26:52.365381884Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
|
| 53 |
+
{"time":"2026-07-23T01:26:53.365610409Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
|
| 54 |
+
{"time":"2026-07-23T01:26:53.365639983Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
|
| 55 |
+
{"time":"2026-07-23T01:26:53.365656633Z","level":"INFO","msg":"server is closed"}
|
| 56 |
+
{"time":"2026-07-23T01:26:53.378096243Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
|
| 57 |
+
{"time":"2026-07-23T01:26:53.378116102Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
|
| 58 |
+
{"time":"2026-07-23T01:26:53.37812429Z","level":"INFO","msg":"server is closed"}
|
| 59 |
+
{"time":"2026-07-23T01:26:53.728940335Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
|
| 60 |
+
{"time":"2026-07-23T01:26:53.728959563Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
|
| 61 |
+
{"time":"2026-07-23T01:26:53.728969099Z","level":"INFO","msg":"server is closed"}
|
| 62 |
+
{"time":"2026-07-23T01:27:02.095210277Z","level":"INFO","msg":"server: parent process exited, terminating service process"}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug-internal.log
ADDED
|
@@ -0,0 +1,185 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{"time":"2026-07-23T00:58:50.67407502Z","level":"INFO","msg":"wandb-core"}
|
| 2 |
+
{"time":"2026-07-23T00:58:50.674523567Z","level":"INFO","msg":"stream: starting","core version":"0.27.2"}
|
| 3 |
+
{"time":"2026-07-23T00:58:51.13116946Z","level":"INFO","msg":"stream: created new stream","id":"21sn30p5"}
|
| 4 |
+
{"time":"2026-07-23T00:58:51.131340942Z","level":"INFO","msg":"stream: started"}
|
| 5 |
+
{"time":"2026-07-23T00:58:51.131374554Z","level":"INFO","msg":"handler: started"}
|
| 6 |
+
{"time":"2026-07-23T00:58:51.131422604Z","level":"INFO","msg":"writer: started","stream_id":"21sn30p5"}
|
| 7 |
+
{"time":"2026-07-23T00:58:51.131436535Z","level":"INFO","msg":"sender: started"}
|
| 8 |
+
{"time":"2026-07-23T00:58:51.827561847Z","level":"INFO","msg":"filestream: sending request","total_files":1}
|
| 9 |
+
{"time":"2026-07-23T00:59:08.297425216Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 10 |
+
{"time":"2026-07-23T00:59:08.297654843Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":0,"events_lines":2,"console_offset":0,"console_lines":30,"uploaded_len":2}
|
| 11 |
+
{"time":"2026-07-23T00:59:37.251300004Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 12 |
+
{"time":"2026-07-23T00:59:37.25155055Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":2,"events_lines":4,"console_offset":30,"console_lines":11}
|
| 13 |
+
{"time":"2026-07-23T01:00:00.774437347Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 14 |
+
{"time":"2026-07-23T01:00:00.774644628Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":6,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 15 |
+
{"time":"2026-07-23T01:00:17.348507832Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 16 |
+
{"time":"2026-07-23T01:00:17.34872665Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":8,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 17 |
+
{"time":"2026-07-23T01:00:18.082699427Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 18 |
+
{"time":"2026-07-23T01:00:18.318755789Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":40,"console_lines":1}
|
| 19 |
+
{"time":"2026-07-23T01:00:19.033204319Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 20 |
+
{"time":"2026-07-23T01:00:33.319449994Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":10,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 21 |
+
{"time":"2026-07-23T01:00:33.73029968Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 22 |
+
{"time":"2026-07-23T01:00:48.319361819Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":12,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 23 |
+
{"time":"2026-07-23T01:00:48.961620273Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 24 |
+
{"time":"2026-07-23T01:01:03.319045011Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":14,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 25 |
+
{"time":"2026-07-23T01:01:54.913746436Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 26 |
+
{"time":"2026-07-23T01:01:54.913952832Z","level":"INFO","msg":"filestream: sending request","total_files":0}
|
| 27 |
+
{"time":"2026-07-23T01:01:55.711048389Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 28 |
+
{"time":"2026-07-23T01:01:55.71128712Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":16,"events_lines":8,"console_offset":40,"console_lines":1}
|
| 29 |
+
{"time":"2026-07-23T01:01:56.772495513Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 30 |
+
{"time":"2026-07-23T01:01:56.772649169Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":40,"console_lines":1}
|
| 31 |
+
{"time":"2026-07-23T01:01:57.690015841Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 32 |
+
{"time":"2026-07-23T01:02:10.640099044Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":24,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 33 |
+
{"time":"2026-07-23T01:02:36.316166553Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 34 |
+
{"time":"2026-07-23T01:02:36.316391302Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":26,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 35 |
+
{"time":"2026-07-23T01:02:36.577308081Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 36 |
+
{"time":"2026-07-23T01:02:40.640139406Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":28,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 37 |
+
{"time":"2026-07-23T01:02:40.998944203Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 38 |
+
{"time":"2026-07-23T01:02:55.640156821Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":30,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 39 |
+
{"time":"2026-07-23T01:03:02.304763647Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 40 |
+
{"time":"2026-07-23T01:03:10.640151405Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":32,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 41 |
+
{"time":"2026-07-23T01:03:17.839821438Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 42 |
+
{"time":"2026-07-23T01:03:25.639769825Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":34,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 43 |
+
{"time":"2026-07-23T01:03:31.067611483Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 44 |
+
{"time":"2026-07-23T01:03:40.640535564Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":36,"events_lines":2,"console_offset":40,"console_lines":1}
|
| 45 |
+
{"time":"2026-07-23T01:03:42.028352221Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 46 |
+
{"time":"2026-07-23T01:03:51.78175928Z","level":"WARN","msg":"handler: ignoring partial history record","step":30200,"current":37801}
|
| 47 |
+
{"time":"2026-07-23T01:03:55.639549937Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":38,"events_lines":2,"console_offset":40,"console_lines":2}
|
| 48 |
+
{"time":"2026-07-23T01:04:29.041072512Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 49 |
+
{"time":"2026-07-23T01:04:29.041195899Z","level":"INFO","msg":"filestream: sending request","total_files":0}
|
| 50 |
+
{"time":"2026-07-23T01:04:36.20151358Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 51 |
+
{"time":"2026-07-23T01:04:36.201702259Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":40,"events_lines":4,"console_offset":41,"console_lines":1}
|
| 52 |
+
{"time":"2026-07-23T01:04:36.525501449Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 53 |
+
{"time":"2026-07-23T01:04:36.526104662Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":41,"console_lines":1}
|
| 54 |
+
{"time":"2026-07-23T01:04:36.963517774Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 55 |
+
{"time":"2026-07-23T01:04:44.195640018Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":44,"events_lines":2,"console_offset":41,"console_lines":1}
|
| 56 |
+
{"time":"2026-07-23T01:05:18.030361841Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 57 |
+
{"time":"2026-07-23T01:05:18.030509078Z","level":"INFO","msg":"filestream: sending request","total_files":0}
|
| 58 |
+
{"time":"2026-07-23T01:05:18.482621576Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 59 |
+
{"time":"2026-07-23T01:05:18.482843557Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":46,"events_lines":4,"console_offset":41,"console_lines":1}
|
| 60 |
+
{"time":"2026-07-23T01:05:45.521040573Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 61 |
+
{"time":"2026-07-23T01:05:45.521264933Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":41,"console_lines":1}
|
| 62 |
+
{"time":"2026-07-23T01:05:46.34078666Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 63 |
+
{"time":"2026-07-23T01:05:46.341113512Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":50,"events_lines":4,"console_offset":41,"console_lines":1}
|
| 64 |
+
{"time":"2026-07-23T01:06:20.967979378Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 65 |
+
{"time":"2026-07-23T01:06:20.968143796Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":41,"console_lines":1}
|
| 66 |
+
{"time":"2026-07-23T01:06:21.788861644Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 67 |
+
{"time":"2026-07-23T01:06:21.78911254Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":54,"events_lines":4,"console_offset":41,"console_lines":1}
|
| 68 |
+
{"time":"2026-07-23T01:06:22.60665294Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 69 |
+
{"time":"2026-07-23T01:06:22.606811084Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":41,"console_lines":1}
|
| 70 |
+
{"time":"2026-07-23T01:06:52.55059983Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 71 |
+
{"time":"2026-07-23T01:06:52.550770901Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":58,"events_lines":2,"console_offset":41,"console_lines":1}
|
| 72 |
+
{"time":"2026-07-23T01:06:54.005857754Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 73 |
+
{"time":"2026-07-23T01:06:54.006028608Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":60,"events_lines":4,"console_offset":41,"console_lines":1}
|
| 74 |
+
{"time":"2026-07-23T01:07:17.537314015Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 75 |
+
{"time":"2026-07-23T01:07:17.537456039Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":41,"console_lines":1}
|
| 76 |
+
{"time":"2026-07-23T01:07:19.511331988Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 77 |
+
{"time":"2026-07-23T01:07:21.719190633Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":64,"events_lines":2,"console_offset":41,"console_lines":1}
|
| 78 |
+
{"time":"2026-07-23T01:07:55.792428434Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 79 |
+
{"time":"2026-07-23T01:07:55.792602686Z","level":"INFO","msg":"filestream: sending request","total_files":0}
|
| 80 |
+
{"time":"2026-07-23T01:08:16.271801254Z","level":"WARN","msg":"handler: ignoring partial history record","step":30400,"current":37801}
|
| 81 |
+
{"time":"2026-07-23T01:08:18.122096142Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 82 |
+
{"time":"2026-07-23T01:08:18.122286759Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":66,"events_lines":6,"console_offset":41,"console_lines":1}
|
| 83 |
+
{"time":"2026-07-23T01:08:43.134370157Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 84 |
+
{"time":"2026-07-23T01:08:43.134590054Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":72,"events_lines":2,"console_offset":41,"console_lines":2}
|
| 85 |
+
{"time":"2026-07-23T01:09:18.570707037Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 86 |
+
{"time":"2026-07-23T01:09:18.570914275Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":74,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 87 |
+
{"time":"2026-07-23T01:09:41.888855012Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 88 |
+
{"time":"2026-07-23T01:09:41.889196416Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":78,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 89 |
+
{"time":"2026-07-23T01:09:58.349681222Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 90 |
+
{"time":"2026-07-23T01:09:58.349864944Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":82,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 91 |
+
{"time":"2026-07-23T01:10:32.298562655Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 92 |
+
{"time":"2026-07-23T01:10:32.298733994Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":86,"events_lines":2,"console_offset":42,"console_lines":1}
|
| 93 |
+
{"time":"2026-07-23T01:10:47.75234985Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 94 |
+
{"time":"2026-07-23T01:10:47.752530378Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":88,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 95 |
+
{"time":"2026-07-23T01:10:50.304345242Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 96 |
+
{"time":"2026-07-23T01:10:50.30450987Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":92,"events_lines":2,"console_offset":42,"console_lines":1}
|
| 97 |
+
{"time":"2026-07-23T01:10:53.87076098Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 98 |
+
{"time":"2026-07-23T01:10:53.870915944Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":42,"console_lines":1}
|
| 99 |
+
{"time":"2026-07-23T01:11:03.880283481Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 100 |
+
{"time":"2026-07-23T01:11:03.880585986Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":94,"events_lines":2,"console_offset":42,"console_lines":1}
|
| 101 |
+
{"time":"2026-07-23T01:11:47.361493816Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 102 |
+
{"time":"2026-07-23T01:11:47.361663342Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":96,"events_lines":2,"console_offset":42,"console_lines":1}
|
| 103 |
+
{"time":"2026-07-23T01:12:21.015656975Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 104 |
+
{"time":"2026-07-23T01:12:21.015941791Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":98,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 105 |
+
{"time":"2026-07-23T01:12:23.637694617Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 106 |
+
{"time":"2026-07-23T01:12:23.637945315Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":102,"events_lines":4,"console_offset":42,"console_lines":1}
|
| 107 |
+
{"time":"2026-07-23T01:12:26.18324697Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 108 |
+
{"time":"2026-07-23T01:12:26.183484773Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":106,"events_lines":2,"console_offset":42,"console_lines":1}
|
| 109 |
+
{"time":"2026-07-23T01:12:39.916375063Z","level":"WARN","msg":"handler: ignoring partial history record","step":30600,"current":37801}
|
| 110 |
+
{"time":"2026-07-23T01:12:49.940931312Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 111 |
+
{"time":"2026-07-23T01:12:49.94106747Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":42,"console_lines":1}
|
| 112 |
+
{"time":"2026-07-23T01:13:23.643258289Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 113 |
+
{"time":"2026-07-23T01:13:23.643462897Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":108,"events_lines":2,"console_offset":42,"console_lines":2}
|
| 114 |
+
{"time":"2026-07-23T01:14:07.777631625Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 115 |
+
{"time":"2026-07-23T01:14:07.777909415Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":110,"events_lines":6,"console_offset":43,"console_lines":1}
|
| 116 |
+
{"time":"2026-07-23T01:14:48.055036754Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 117 |
+
{"time":"2026-07-23T01:14:48.055232649Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":116,"events_lines":6,"console_offset":43,"console_lines":1}
|
| 118 |
+
{"time":"2026-07-23T01:14:57.255096375Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 119 |
+
{"time":"2026-07-23T01:14:57.255261696Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":122,"events_lines":4,"console_offset":43,"console_lines":1}
|
| 120 |
+
{"time":"2026-07-23T01:15:31.605211554Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 121 |
+
{"time":"2026-07-23T01:15:31.605388571Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":126,"events_lines":2,"console_offset":43,"console_lines":1}
|
| 122 |
+
{"time":"2026-07-23T01:16:05.215922359Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 123 |
+
{"time":"2026-07-23T01:16:05.216180035Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":128,"events_lines":4,"console_offset":43,"console_lines":1}
|
| 124 |
+
{"time":"2026-07-23T01:16:49.69634125Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 125 |
+
{"time":"2026-07-23T01:16:49.696584322Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":132,"events_lines":4,"console_offset":43,"console_lines":1}
|
| 126 |
+
{"time":"2026-07-23T01:17:16.154985994Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 127 |
+
{"time":"2026-07-23T01:17:16.155238543Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":136,"events_lines":6,"console_offset":43,"console_lines":1}
|
| 128 |
+
{"time":"2026-07-23T01:17:19.873968641Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 129 |
+
{"time":"2026-07-23T01:17:19.874170612Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":142,"events_lines":4,"console_offset":43,"console_lines":1}
|
| 130 |
+
{"time":"2026-07-23T01:17:23.592800748Z","level":"WARN","msg":"handler: ignoring partial history record","step":30800,"current":37801}
|
| 131 |
+
{"time":"2026-07-23T01:17:47.395698123Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 132 |
+
{"time":"2026-07-23T01:17:47.395863178Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":43,"console_lines":1}
|
| 133 |
+
{"time":"2026-07-23T01:18:41.196027148Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 134 |
+
{"time":"2026-07-23T01:18:41.196305281Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":146,"events_lines":4,"console_offset":43,"console_lines":2}
|
| 135 |
+
{"time":"2026-07-23T01:18:44.967501709Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 136 |
+
{"time":"2026-07-23T01:18:44.967785704Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":150,"events_lines":8,"console_offset":44,"console_lines":1}
|
| 137 |
+
{"time":"2026-07-23T01:19:09.096940616Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 138 |
+
{"time":"2026-07-23T01:19:09.097100102Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":44,"console_lines":1}
|
| 139 |
+
{"time":"2026-07-23T01:19:39.774502906Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 140 |
+
{"time":"2026-07-23T01:19:39.774716901Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":158,"events_lines":4,"console_offset":44,"console_lines":1}
|
| 141 |
+
{"time":"2026-07-23T01:20:04.896487412Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 142 |
+
{"time":"2026-07-23T01:20:04.898339285Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":162,"events_lines":4,"console_offset":44,"console_lines":1}
|
| 143 |
+
{"time":"2026-07-23T01:20:05.387935141Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 144 |
+
{"time":"2026-07-23T01:20:05.388096529Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":166,"events_lines":2,"console_offset":44,"console_lines":1}
|
| 145 |
+
{"time":"2026-07-23T01:20:05.80043589Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 146 |
+
{"time":"2026-07-23T01:20:05.866398254Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":44,"console_lines":1}
|
| 147 |
+
{"time":"2026-07-23T01:20:06.254596059Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 148 |
+
{"time":"2026-07-23T01:20:20.867279447Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":168,"events_lines":2,"console_offset":44,"console_lines":1}
|
| 149 |
+
{"time":"2026-07-23T01:20:21.313876641Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 150 |
+
{"time":"2026-07-23T01:20:35.867038496Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":170,"events_lines":2,"console_offset":44,"console_lines":1}
|
| 151 |
+
{"time":"2026-07-23T01:20:36.380081928Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 152 |
+
{"time":"2026-07-23T01:20:50.867254226Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":172,"events_lines":2,"console_offset":44,"console_lines":1}
|
| 153 |
+
{"time":"2026-07-23T01:21:32.123439622Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 154 |
+
{"time":"2026-07-23T01:21:32.123634428Z","level":"INFO","msg":"filestream: sending request","total_files":0}
|
| 155 |
+
{"time":"2026-07-23T01:21:32.580253054Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 156 |
+
{"time":"2026-07-23T01:21:32.580510985Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":174,"events_lines":6,"console_offset":44,"console_lines":1}
|
| 157 |
+
{"time":"2026-07-23T01:21:52.507621458Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 158 |
+
{"time":"2026-07-23T01:21:52.507858451Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":44,"console_lines":1}
|
| 159 |
+
{"time":"2026-07-23T01:21:52.899822317Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 160 |
+
{"time":"2026-07-23T01:21:52.90008305Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":180,"events_lines":4,"console_offset":44,"console_lines":1}
|
| 161 |
+
{"time":"2026-07-23T01:22:15.448208736Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 162 |
+
{"time":"2026-07-23T01:22:15.448412109Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":44,"console_lines":1}
|
| 163 |
+
{"time":"2026-07-23T01:22:47.174396816Z","level":"WARN","msg":"handler: ignoring partial history record","step":31000,"current":37801}
|
| 164 |
+
{"time":"2026-07-23T01:22:54.565274455Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 165 |
+
{"time":"2026-07-23T01:22:54.56545658Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":184,"events_lines":2,"console_offset":44,"console_lines":1}
|
| 166 |
+
{"time":"2026-07-23T01:22:55.114516539Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 167 |
+
{"time":"2026-07-23T01:22:55.115149232Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":186,"events_lines":6,"console_offset":44,"console_lines":2}
|
| 168 |
+
{"time":"2026-07-23T01:23:46.222934062Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 169 |
+
{"time":"2026-07-23T01:23:46.223228008Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":45,"console_lines":1}
|
| 170 |
+
{"time":"2026-07-23T01:24:17.987010297Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 171 |
+
{"time":"2026-07-23T01:24:17.987300877Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":192,"events_lines":6,"console_offset":45,"console_lines":1}
|
| 172 |
+
{"time":"2026-07-23T01:25:14.919547107Z","level":"INFO","msg":"api: retrying HTTP error","status":500,"url":"https://api.wandb.ai/files/minje227_hyu-hanyang-university/lerobot/21sn30p5/file_stream","body":"{\"error\":\"context deadline exceeded\"}"}
|
| 173 |
+
{"time":"2026-07-23T01:25:33.395684774Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 174 |
+
{"time":"2026-07-23T01:25:33.39616829Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":198,"events_lines":4,"console_offset":45,"console_lines":1}
|
| 175 |
+
{"time":"2026-07-23T01:26:03.938141908Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 176 |
+
{"time":"2026-07-23T01:26:03.938596709Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":202,"events_lines":10,"console_offset":45,"console_lines":1}
|
| 177 |
+
{"time":"2026-07-23T01:26:04.698872028Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 178 |
+
{"time":"2026-07-23T01:26:04.699164454Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":212,"events_lines":4,"console_offset":45,"console_lines":1}
|
| 179 |
+
{"time":"2026-07-23T01:26:05.542834304Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 180 |
+
{"time":"2026-07-23T01:26:05.54305129Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":45,"console_lines":1}
|
| 181 |
+
{"time":"2026-07-23T01:26:06.2451229Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 182 |
+
{"time":"2026-07-23T01:26:20.065201503Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":216,"events_lines":2,"console_offset":45,"console_lines":1}
|
| 183 |
+
{"time":"2026-07-23T01:26:20.516128671Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
|
| 184 |
+
{"time":"2026-07-23T01:26:35.065455634Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":218,"events_lines":2,"console_offset":45,"console_lines":1}
|
| 185 |
+
{"time":"2026-07-23T01:26:53.032064162Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug.log
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
|
| 2 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_setup.py:_flush():81] Configure stats pid to 430
|
| 3 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug.log
|
| 5 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_005850-21sn30p5/logs/debug-internal.log
|
| 6 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_init.py:init():766] calling init triggers
|
| 7 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [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': [10]}}, '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/Abs_6D/DiT_Layer/8/Baseline/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': None, '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, 'action_expert_num_layers': 8, 'rkd_enabled': False, 'rkd_vae_checkpoint': None, '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': 0.1, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 1.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/Baseline', 'job_name': 'INSIGHT_6D_DiT8_Baseline', 'resume': True, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 64, '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': '21sn30p5', '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/Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000', '_wandb': {}}
|
| 9 |
+
2026-07-23 00:58:50,452 INFO MainThread:430 [wandb_init.py:init():814] starting backend
|
| 10 |
+
2026-07-23 00:58:50,668 INFO MainThread:430 [wandb_init.py:init():829] sending inform_init request
|
| 11 |
+
2026-07-23 00:58:51,132 INFO MainThread:430 [wandb_init.py:init():834] backend started and connected
|
| 12 |
+
2026-07-23 00:58:51,134 INFO MainThread:430 [wandb_init.py:init():904] updated telemetry
|
| 13 |
+
2026-07-23 00:58:51,138 INFO MainThread:430 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-07-23 00:58:51,824 INFO MainThread:430 [wandb_init.py:init():967] run resumed
|
| 15 |
+
2026-07-23 00:58:51,826 INFO MainThread:430 [wandb_init.py:init():972] starting run threads in backend
|
| 16 |
+
2026-07-23 00:58:51,878 INFO MainThread:430 [wandb_run.py:_console_start():2523] atexit reg
|
| 17 |
+
2026-07-23 00:58:51,878 INFO MainThread:430 [wandb_run.py:_redirect():2373] redirect: wrap_raw
|
| 18 |
+
2026-07-23 00:58:51,879 INFO MainThread:430 [wandb_run.py:_redirect():2442] Wrapping output streams.
|
| 19 |
+
2026-07-23 00:58:51,879 INFO MainThread:430 [wandb_run.py:_redirect():2465] Redirects installed.
|
| 20 |
+
2026-07-23 00:58:51,881 INFO MainThread:430 [wandb_init.py:init():1010] run started, returning control to user process
|
| 21 |
+
2026-07-23 01:26:52,365 INFO wandb-AsyncioManager-main:430 [service_client.py:_forward_responses():122] Reached EOF.
|
| 22 |
+
2026-07-23 01:26:52,365 INFO wandb-AsyncioManager-main:430 [mailbox.py:close():154] Closing mailbox, abandoning 2 handles.
|
| 23 |
+
2026-07-23 01:26:52,365 ERROR wandb-AsyncioManager-main:430 [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/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/files/output.log
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO 2026-07-23 01:33:43 db_utils.py:121 Logs will be synced with wandb.
|
| 2 |
+
INFO 2026-07-23 01:33:43 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/21sn30p5
|
| 3 |
+
INFO 2026-07-23 01:33:43 ot_train.py:298 Creating dataset
|
| 4 |
+
INFO 2026-07-23 01:33:45 ot_train.py:332 Creating policy
|
| 5 |
+
INFO 2026-07-23 01:33:45 ng_groot.py:193 The Groot policy wraps NVIDIA's GR00T n1.7 model. Loading pretrained model from: /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline/checkpoints/030000/pretrained_model
|
| 6 |
+
INFO 2026-07-23 01:33:45 ng_groot.py:227 Detected fine-tuned LeRobot checkpoint, loading with state dict...
|
| 7 |
+
Fetching 27 files: 100%|ββββββββββ| 27/27 [00:00<00:00, 419.61it/s]
|
| 8 |
+
`torch_dtype` is deprecated! Use `dtype` instead!
|
| 9 |
+
Loading weights: 100%|ββββββββββ| 695/695 [00:01<00:00, 425.50it/s]
|
| 10 |
+
[1mGR00TN17 LOAD REPORT[0m from: /home/ext_minje/.cache/huggingface/hub/models--nvidia--GR00T-N1.7-3B/snapshots/2fc962b973bccdd5d8ce4f67cc63b264d6886495
|
| 11 |
+
Key | Status | |
|
| 12 |
+
--------------------------------------------------------------------+------------+--+-
|
| 13 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.bias | UNEXPECTED | |
|
| 14 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.bias | UNEXPECTED | |
|
| 15 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.bias | UNEXPECTED | |
|
| 16 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.weight | UNEXPECTED | |
|
| 17 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.weight | UNEXPECTED | |
|
| 18 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.weight | UNEXPECTED | |
|
| 19 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.bias | UNEXPECTED | |
|
| 20 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.bias | UNEXPECTED | |
|
| 21 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.weight | UNEXPECTED | |
|
| 22 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.weight | UNEXPECTED | |
|
| 23 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.bias | UNEXPECTED | |
|
| 24 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.weight | UNEXPECTED | |
|
| 25 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.bias | UNEXPECTED | |
|
| 26 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.weight | UNEXPECTED | |
|
| 27 |
+
|
| 28 |
+
Notes:
|
| 29 |
+
- UNEXPECTED: can be ignored when loading from different task/architecture; not ok if you expect identical arch.
|
| 30 |
+
Loading weights from local directory
|
| 31 |
+
INFO 2026-07-23 01:34:05 ot_train.py:405 Creating optimizer and scheduler
|
| 32 |
+
INFO 2026-07-23 01:34:07 ot_train.py:437 Output dir: /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline
|
| 33 |
+
INFO 2026-07-23 01:34:07 ot_train.py:444 cfg.steps=60000 (60K)
|
| 34 |
+
INFO 2026-07-23 01:34:07 ot_train.py:445 dataset.num_frames=218367 (218K)
|
| 35 |
+
INFO 2026-07-23 01:34:07 ot_train.py:446 dataset.num_episodes=4930
|
| 36 |
+
INFO 2026-07-23 01:34:07 ot_train.py:449 Effective batch size: 64 x 1 = 64
|
| 37 |
+
INFO 2026-07-23 01:34:07 ot_train.py:450 num_learnable_params=808512640 (809M)
|
| 38 |
+
INFO 2026-07-23 01:34:07 ot_train.py:451 num_total_params=2333587585 (2B)
|
| 39 |
+
INFO 2026-07-23 01:34:07 ot_train.py:492 Resuming data order at epoch 13, sample 42176
|
| 40 |
+
Training: 0%| | 0/30000 [00:00<?, ?step/s]INFO 2026-07-23 01:34:08 ot_train.py:602 Start offline training on a fixed dataset, with effective batch size: 64
|
| 41 |
+
Training: 0%| | 46/30000 [01:57<19:41:45, 2.37s/step]
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/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/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug.log
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Current SDK version is 0.27.2
|
| 2 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Configure stats pid to 814701
|
| 3 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug.log
|
| 5 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/logs/debug-internal.log
|
| 6 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:init():766] calling init triggers
|
| 7 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [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': [10]}}, '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/DiT_Layer/8/Baseline/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': None, '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, 'action_expert_num_layers': 8, 'rkd_enabled': False, 'rkd_vae_checkpoint': None, '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': 0.1, 'rkd_distance_weight': 1.0, 'rkd_angle_weight': 1.0}, 'reward_model': None, 'output_dir': '/home/ext_minje/groot_insight/train_ckpt/Abs_6D/DiT_Layer/8/Baseline', 'job_name': 'INSIGHT_6D_DiT8_Baseline', 'resume': True, 'seed': 42, 'cudnn_deterministic': False, 'num_workers': 4, 'batch_size': 64, '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': '21sn30p5', '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/DiT_Layer/8/Baseline/checkpoints/030000', '_wandb': {}}
|
| 9 |
+
2026-07-23 01:33:42,309 INFO MainThread:814701 [wandb_init.py:init():814] starting backend
|
| 10 |
+
2026-07-23 01:33:42,527 INFO MainThread:814701 [wandb_init.py:init():829] sending inform_init request
|
| 11 |
+
2026-07-23 01:33:42,987 INFO MainThread:814701 [wandb_init.py:init():834] backend started and connected
|
| 12 |
+
2026-07-23 01:33:42,989 INFO MainThread:814701 [wandb_init.py:init():904] updated telemetry
|
| 13 |
+
2026-07-23 01:33:42,993 INFO MainThread:814701 [wandb_init.py:init():927] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-07-23 01:33:43,715 INFO MainThread:814701 [wandb_init.py:init():967] run resumed
|
| 15 |
+
2026-07-23 01:33:43,718 INFO MainThread:814701 [wandb_init.py:init():972] starting run threads in backend
|
| 16 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_console_start():2523] atexit reg
|
| 17 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2373] redirect: wrap_raw
|
| 18 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2442] Wrapping output streams.
|
| 19 |
+
2026-07-23 01:33:43,819 INFO MainThread:814701 [wandb_run.py:_redirect():2465] Redirects installed.
|
| 20 |
+
2026-07-23 01:33:43,821 INFO MainThread:814701 [wandb_init.py:init():1010] run started, returning control to user process
|
Abs_6D/DiT_Layer/8/Baseline/wandb/run-20260723_013342-21sn30p5/run-21sn30p5.wandb
ADDED
|
Binary file (32.8 kB). View file
|
|
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/config.json
ADDED
|
@@ -0,0 +1,118 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": 3000,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 109 |
+
"rkd_enabled": true,
|
| 110 |
+
"rkd_vae_checkpoint": "/home/ext_minje/groot_insight/TimewarpVAE_ckpt/TimewarpVAE_combo_top8_30k/C8_01_C3_008_beta_5em05_latent_dim_16_granularity_4_decoder_hidden_dim_512/checkpoint-step-30000-epoch-39.11",
|
| 111 |
+
"rkd_student_pooling": "attention",
|
| 112 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 113 |
+
"rkd_student_attention_output_dim": 512,
|
| 114 |
+
"rkd_fm_loss_weight": 1.0,
|
| 115 |
+
"rkd_loss_weight": 1.0,
|
| 116 |
+
"rkd_distance_weight": 1.0,
|
| 117 |
+
"rkd_angle_weight": 2.0
|
| 118 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/phase_schedule.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "rscl_rkd_cosine_decay_60K",
|
| 11 |
+
"start_step": 0,
|
| 12 |
+
"until_step": 60000,
|
| 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": "cosine",
|
| 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 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/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": 10,
|
| 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/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/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/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/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/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/pretrained_model/train_config.json
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": 3000,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 194 |
+
"rkd_enabled": true,
|
| 195 |
+
"rkd_vae_checkpoint": "/home/ext_minje/groot_insight/TimewarpVAE_ckpt/TimewarpVAE_combo_top8_30k/C8_01_C3_008_beta_5em05_latent_dim_16_granularity_4_decoder_hidden_dim_512/checkpoint-step-30000-epoch-39.11",
|
| 196 |
+
"rkd_student_pooling": "attention",
|
| 197 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 198 |
+
"rkd_student_attention_output_dim": 512,
|
| 199 |
+
"rkd_fm_loss_weight": 1.0,
|
| 200 |
+
"rkd_loss_weight": 1.0,
|
| 201 |
+
"rkd_distance_weight": 1.0,
|
| 202 |
+
"rkd_angle_weight": 2.0
|
| 203 |
+
},
|
| 204 |
+
"reward_model": null,
|
| 205 |
+
"output_dir": "/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K",
|
| 206 |
+
"job_name": "INSIGHT_6D_DiT8_RKD_TimewarpVAE_RSCLstyle_C60",
|
| 207 |
+
"resume": false,
|
| 208 |
+
"seed": 42,
|
| 209 |
+
"cudnn_deterministic": false,
|
| 210 |
+
"num_workers": 4,
|
| 211 |
+
"batch_size": 64,
|
| 212 |
+
"prefetch_factor": 4,
|
| 213 |
+
"persistent_workers": true,
|
| 214 |
+
"steps": 60000,
|
| 215 |
+
"env_eval_freq": 20000,
|
| 216 |
+
"log_freq": 200,
|
| 217 |
+
"eval_steps": 0,
|
| 218 |
+
"max_eval_samples": 0,
|
| 219 |
+
"tolerance_s": 0.0001,
|
| 220 |
+
"save_checkpoint": true,
|
| 221 |
+
"save_freq": 20000,
|
| 222 |
+
"use_policy_training_preset": true,
|
| 223 |
+
"optimizer": {
|
| 224 |
+
"type": "adamw",
|
| 225 |
+
"lr": 0.0001,
|
| 226 |
+
"weight_decay": 1e-05,
|
| 227 |
+
"grad_clip_norm": 1.0,
|
| 228 |
+
"betas": [
|
| 229 |
+
0.9,
|
| 230 |
+
0.999
|
| 231 |
+
],
|
| 232 |
+
"eps": 1e-08
|
| 233 |
+
},
|
| 234 |
+
"scheduler": {
|
| 235 |
+
"type": "diffuser",
|
| 236 |
+
"num_warmup_steps": 3000,
|
| 237 |
+
"name": "cosine"
|
| 238 |
+
},
|
| 239 |
+
"eval": {
|
| 240 |
+
"n_episodes": 50,
|
| 241 |
+
"batch_size": 50,
|
| 242 |
+
"use_async_envs": true,
|
| 243 |
+
"recording": false,
|
| 244 |
+
"recording_repo_id": null,
|
| 245 |
+
"recording_private": false
|
| 246 |
+
},
|
| 247 |
+
"wandb": {
|
| 248 |
+
"enable": true,
|
| 249 |
+
"disable_artifact": true,
|
| 250 |
+
"project": "lerobot",
|
| 251 |
+
"entity": null,
|
| 252 |
+
"notes": null,
|
| 253 |
+
"run_id": "rd8tz18f",
|
| 254 |
+
"mode": null,
|
| 255 |
+
"add_tags": true
|
| 256 |
+
},
|
| 257 |
+
"peft": null,
|
| 258 |
+
"job": {
|
| 259 |
+
"target": null,
|
| 260 |
+
"image": "huggingface/lerobot-gpu:latest",
|
| 261 |
+
"timeout": "2d",
|
| 262 |
+
"detach": false,
|
| 263 |
+
"tags": []
|
| 264 |
+
},
|
| 265 |
+
"save_checkpoint_to_hub": false,
|
| 266 |
+
"sample_weighting": null,
|
| 267 |
+
"rename_map": {},
|
| 268 |
+
"checkpoint_path": null
|
| 269 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/training_state/optimizer_param_groups.json
ADDED
|
@@ -0,0 +1,249 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"lr": 9.632470336074009e-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 |
+
]
|
| 123 |
+
},
|
| 124 |
+
{
|
| 125 |
+
"weight_decay": 0.0,
|
| 126 |
+
"lr": 9.632470336074009e-05,
|
| 127 |
+
"betas": [
|
| 128 |
+
0.9,
|
| 129 |
+
0.999
|
| 130 |
+
],
|
| 131 |
+
"eps": 1e-08,
|
| 132 |
+
"amsgrad": false,
|
| 133 |
+
"maximize": false,
|
| 134 |
+
"foreach": null,
|
| 135 |
+
"capturable": false,
|
| 136 |
+
"differentiable": false,
|
| 137 |
+
"fused": null,
|
| 138 |
+
"decoupled_weight_decay": true,
|
| 139 |
+
"initial_lr": 0.0001,
|
| 140 |
+
"params": [
|
| 141 |
+
103,
|
| 142 |
+
104,
|
| 143 |
+
105,
|
| 144 |
+
106,
|
| 145 |
+
107,
|
| 146 |
+
108,
|
| 147 |
+
109,
|
| 148 |
+
110,
|
| 149 |
+
111,
|
| 150 |
+
112,
|
| 151 |
+
113,
|
| 152 |
+
114,
|
| 153 |
+
115,
|
| 154 |
+
116,
|
| 155 |
+
117,
|
| 156 |
+
118,
|
| 157 |
+
119,
|
| 158 |
+
120,
|
| 159 |
+
121,
|
| 160 |
+
122,
|
| 161 |
+
123,
|
| 162 |
+
124,
|
| 163 |
+
125,
|
| 164 |
+
126,
|
| 165 |
+
127,
|
| 166 |
+
128,
|
| 167 |
+
129,
|
| 168 |
+
130,
|
| 169 |
+
131,
|
| 170 |
+
132,
|
| 171 |
+
133,
|
| 172 |
+
134,
|
| 173 |
+
135,
|
| 174 |
+
136,
|
| 175 |
+
137,
|
| 176 |
+
138,
|
| 177 |
+
139,
|
| 178 |
+
140,
|
| 179 |
+
141,
|
| 180 |
+
142,
|
| 181 |
+
143,
|
| 182 |
+
144,
|
| 183 |
+
145,
|
| 184 |
+
146,
|
| 185 |
+
147,
|
| 186 |
+
148,
|
| 187 |
+
149,
|
| 188 |
+
150,
|
| 189 |
+
151,
|
| 190 |
+
152,
|
| 191 |
+
153,
|
| 192 |
+
154,
|
| 193 |
+
155,
|
| 194 |
+
156,
|
| 195 |
+
157,
|
| 196 |
+
158,
|
| 197 |
+
159,
|
| 198 |
+
160,
|
| 199 |
+
161,
|
| 200 |
+
162,
|
| 201 |
+
163,
|
| 202 |
+
164,
|
| 203 |
+
165,
|
| 204 |
+
166,
|
| 205 |
+
167,
|
| 206 |
+
168,
|
| 207 |
+
169,
|
| 208 |
+
170,
|
| 209 |
+
171,
|
| 210 |
+
172,
|
| 211 |
+
173,
|
| 212 |
+
174,
|
| 213 |
+
175,
|
| 214 |
+
176,
|
| 215 |
+
177,
|
| 216 |
+
178,
|
| 217 |
+
179,
|
| 218 |
+
180,
|
| 219 |
+
181,
|
| 220 |
+
182,
|
| 221 |
+
183,
|
| 222 |
+
184,
|
| 223 |
+
185,
|
| 224 |
+
186,
|
| 225 |
+
187,
|
| 226 |
+
188,
|
| 227 |
+
189,
|
| 228 |
+
190,
|
| 229 |
+
191,
|
| 230 |
+
192,
|
| 231 |
+
193,
|
| 232 |
+
194,
|
| 233 |
+
195,
|
| 234 |
+
196,
|
| 235 |
+
197,
|
| 236 |
+
198,
|
| 237 |
+
199,
|
| 238 |
+
200,
|
| 239 |
+
201,
|
| 240 |
+
202,
|
| 241 |
+
203,
|
| 242 |
+
204,
|
| 243 |
+
205,
|
| 244 |
+
206,
|
| 245 |
+
207,
|
| 246 |
+
208
|
| 247 |
+
]
|
| 248 |
+
}
|
| 249 |
+
]
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000/training_state/scheduler_state.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"base_lrs": [
|
| 3 |
+
0.0001,
|
| 4 |
+
0.0001
|
| 5 |
+
],
|
| 6 |
+
"last_epoch": 10000,
|
| 7 |
+
"_step_count": 10001,
|
| 8 |
+
"_is_initial": false,
|
| 9 |
+
"_get_lr_called_within_step": false,
|
| 10 |
+
"_last_lr": [
|
| 11 |
+
9.632470336074009e-05,
|
| 12 |
+
9.632470336074009e-05
|
| 13 |
+
],
|
| 14 |
+
"lr_lambdas": [
|
| 15 |
+
null,
|
| 16 |
+
null
|
| 17 |
+
]
|
| 18 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/action_space_manifest.json
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 2,
|
| 3 |
+
"action_space": "ee_abs_rot6d",
|
| 4 |
+
"action_views": [
|
| 5 |
+
"ee_abs_rot6d"
|
| 6 |
+
],
|
| 7 |
+
"source_feature": "action.ee_abs_rot6d",
|
| 8 |
+
"canonical_feature": "action",
|
| 9 |
+
"dtype": "float32",
|
| 10 |
+
"shape": [
|
| 11 |
+
10
|
| 12 |
+
],
|
| 13 |
+
"names": [
|
| 14 |
+
"x",
|
| 15 |
+
"y",
|
| 16 |
+
"z",
|
| 17 |
+
"r_col0_x",
|
| 18 |
+
"r_col0_y",
|
| 19 |
+
"r_col0_z",
|
| 20 |
+
"r_col1_x",
|
| 21 |
+
"r_col1_y",
|
| 22 |
+
"r_col1_z",
|
| 23 |
+
"gripper"
|
| 24 |
+
],
|
| 25 |
+
"dataset_repo_id": "Whalswp/INSIGHTfixposV4_filtered_multispace_v2",
|
| 26 |
+
"dataset_revision": "v3.0",
|
| 27 |
+
"dataset_total_frames": 218367,
|
| 28 |
+
"dataset_total_episodes": 4930,
|
| 29 |
+
"dataset_fps": 10,
|
| 30 |
+
"stats_sha256": "71698c7337070aa8267df5aea7ec08bc00d3925510f2649bc20c9201eeb669e4",
|
| 31 |
+
"stats": {
|
| 32 |
+
"min": [
|
| 33 |
+
-0.21784618496894836,
|
| 34 |
+
-0.4474811851978302,
|
| 35 |
+
0.1702737659215927,
|
| 36 |
+
-0.9999986290931702,
|
| 37 |
+
-0.9999991655349731,
|
| 38 |
+
-0.9999999403953552,
|
| 39 |
+
-0.9999990463256836,
|
| 40 |
+
-0.9999998211860657,
|
| 41 |
+
-0.9999988079071045,
|
| 42 |
+
0.0
|
| 43 |
+
],
|
| 44 |
+
"max": [
|
| 45 |
+
0.6283698678016663,
|
| 46 |
+
0.4420371949672699,
|
| 47 |
+
1.051027774810791,
|
| 48 |
+
0.9999999403953552,
|
| 49 |
+
0.9999995231628418,
|
| 50 |
+
1.0,
|
| 51 |
+
0.9999995231628418,
|
| 52 |
+
1.0,
|
| 53 |
+
0.9999996423721313,
|
| 54 |
+
0.03999999910593033
|
| 55 |
+
],
|
| 56 |
+
"mean": [
|
| 57 |
+
0.4040228037735655,
|
| 58 |
+
-0.05247618696989652,
|
| 59 |
+
0.6993459771218912,
|
| 60 |
+
0.07011111991957117,
|
| 61 |
+
-0.03520558904460197,
|
| 62 |
+
0.14470242846953205,
|
| 63 |
+
0.05491166606739672,
|
| 64 |
+
-0.19073058893378805,
|
| 65 |
+
-0.06988117661871228,
|
| 66 |
+
0.021167300266524632
|
| 67 |
+
],
|
| 68 |
+
"std": [
|
| 69 |
+
0.11861927913357073,
|
| 70 |
+
0.11318943557281885,
|
| 71 |
+
0.17849160646987516,
|
| 72 |
+
0.4812528791886511,
|
| 73 |
+
0.7487196258736938,
|
| 74 |
+
0.4251126804030219,
|
| 75 |
+
0.47797229807769137,
|
| 76 |
+
0.6267695346407992,
|
| 77 |
+
0.5782954306815554,
|
| 78 |
+
0.01996590431050817
|
| 79 |
+
],
|
| 80 |
+
"count": [
|
| 81 |
+
218367
|
| 82 |
+
],
|
| 83 |
+
"q01": [
|
| 84 |
+
0.23988881827972064,
|
| 85 |
+
-0.10387461883034811,
|
| 86 |
+
0.6297243923476871,
|
| 87 |
+
-0.43622841955353464,
|
| 88 |
+
-0.6885486527617446,
|
| 89 |
+
-0.14103555388039019,
|
| 90 |
+
-0.4327210063987595,
|
| 91 |
+
-0.9304790915716145,
|
| 92 |
+
-0.3465462258490417,
|
| 93 |
+
-1.000000013351432e-10
|
| 94 |
+
],
|
| 95 |
+
"q10": [
|
| 96 |
+
0.3042306436755216,
|
| 97 |
+
-0.09609991620622624,
|
| 98 |
+
0.6394063072781145,
|
| 99 |
+
-0.3351306352998669,
|
| 100 |
+
-0.5751963015002924,
|
| 101 |
+
-0.08490621283307866,
|
| 102 |
+
-0.33849701676224275,
|
| 103 |
+
-0.8340557712529717,
|
| 104 |
+
-0.29760290790884514,
|
| 105 |
+
-1.000000013351432e-10
|
| 106 |
+
],
|
| 107 |
+
"q50": [
|
| 108 |
+
0.4160225263528291,
|
| 109 |
+
-0.06267459152989589,
|
| 110 |
+
0.6811777491060801,
|
| 111 |
+
0.0543536902230377,
|
| 112 |
+
-0.14012251058819217,
|
| 113 |
+
0.04772394280004727,
|
| 114 |
+
0.03838119138716219,
|
| 115 |
+
-0.14865169094195635,
|
| 116 |
+
-0.2073069401172425,
|
| 117 |
+
0.020772594934870346
|
| 118 |
+
],
|
| 119 |
+
"q90": [
|
| 120 |
+
0.46641008755679864,
|
| 121 |
+
0.020233143101890064,
|
| 122 |
+
0.8095135405259374,
|
| 123 |
+
0.48172401592474906,
|
| 124 |
+
0.7398737043070319,
|
| 125 |
+
0.5970841215636613,
|
| 126 |
+
0.4674752875999555,
|
| 127 |
+
0.36281754777672676,
|
| 128 |
+
0.5033361883834496,
|
| 129 |
+
0.039998362213352394
|
| 130 |
+
],
|
| 131 |
+
"q99": [
|
| 132 |
+
0.47320723171849244,
|
| 133 |
+
0.04114262476135705,
|
| 134 |
+
0.8467967259270389,
|
| 135 |
+
0.5867781409980988,
|
| 136 |
+
0.8529584859257228,
|
| 137 |
+
0.7953936054341789,
|
| 138 |
+
0.5768269858732034,
|
| 139 |
+
0.49353349323389695,
|
| 140 |
+
0.7168914440776504,
|
| 141 |
+
0.03999983541667277
|
| 142 |
+
]
|
| 143 |
+
}
|
| 144 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/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": 10,
|
| 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/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/pretrained_model/train_config.json
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
10
|
| 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": 3000,
|
| 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 |
+
"action_expert_num_layers": 8,
|
| 194 |
+
"rkd_enabled": true,
|
| 195 |
+
"rkd_vae_checkpoint": "/home/ext_minje/groot_insight/TimewarpVAE_ckpt/TimewarpVAE_combo_top8_30k/C8_01_C3_008_beta_5em05_latent_dim_16_granularity_4_decoder_hidden_dim_512/checkpoint-step-30000-epoch-39.11",
|
| 196 |
+
"rkd_student_pooling": "attention",
|
| 197 |
+
"rkd_student_attention_hidden_dim": 512,
|
| 198 |
+
"rkd_student_attention_output_dim": 512,
|
| 199 |
+
"rkd_fm_loss_weight": 1.0,
|
| 200 |
+
"rkd_loss_weight": 1.0,
|
| 201 |
+
"rkd_distance_weight": 1.0,
|
| 202 |
+
"rkd_angle_weight": 2.0
|
| 203 |
+
},
|
| 204 |
+
"reward_model": null,
|
| 205 |
+
"output_dir": "/home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K",
|
| 206 |
+
"job_name": "INSIGHT_6D_DiT8_RKD_TimewarpVAE_RSCLstyle_C60",
|
| 207 |
+
"resume": false,
|
| 208 |
+
"seed": 42,
|
| 209 |
+
"cudnn_deterministic": false,
|
| 210 |
+
"num_workers": 4,
|
| 211 |
+
"batch_size": 64,
|
| 212 |
+
"prefetch_factor": 4,
|
| 213 |
+
"persistent_workers": true,
|
| 214 |
+
"steps": 60000,
|
| 215 |
+
"env_eval_freq": 20000,
|
| 216 |
+
"log_freq": 200,
|
| 217 |
+
"eval_steps": 0,
|
| 218 |
+
"max_eval_samples": 0,
|
| 219 |
+
"tolerance_s": 0.0001,
|
| 220 |
+
"save_checkpoint": true,
|
| 221 |
+
"save_freq": 20000,
|
| 222 |
+
"use_policy_training_preset": true,
|
| 223 |
+
"optimizer": {
|
| 224 |
+
"type": "adamw",
|
| 225 |
+
"lr": 0.0001,
|
| 226 |
+
"weight_decay": 1e-05,
|
| 227 |
+
"grad_clip_norm": 1.0,
|
| 228 |
+
"betas": [
|
| 229 |
+
0.9,
|
| 230 |
+
0.999
|
| 231 |
+
],
|
| 232 |
+
"eps": 1e-08
|
| 233 |
+
},
|
| 234 |
+
"scheduler": {
|
| 235 |
+
"type": "diffuser",
|
| 236 |
+
"num_warmup_steps": 3000,
|
| 237 |
+
"name": "cosine"
|
| 238 |
+
},
|
| 239 |
+
"eval": {
|
| 240 |
+
"n_episodes": 50,
|
| 241 |
+
"batch_size": 50,
|
| 242 |
+
"use_async_envs": true,
|
| 243 |
+
"recording": false,
|
| 244 |
+
"recording_repo_id": null,
|
| 245 |
+
"recording_private": false
|
| 246 |
+
},
|
| 247 |
+
"wandb": {
|
| 248 |
+
"enable": true,
|
| 249 |
+
"disable_artifact": true,
|
| 250 |
+
"project": "lerobot",
|
| 251 |
+
"entity": null,
|
| 252 |
+
"notes": null,
|
| 253 |
+
"run_id": "rd8tz18f",
|
| 254 |
+
"mode": null,
|
| 255 |
+
"add_tags": true
|
| 256 |
+
},
|
| 257 |
+
"peft": null,
|
| 258 |
+
"job": {
|
| 259 |
+
"target": null,
|
| 260 |
+
"image": "huggingface/lerobot-gpu:latest",
|
| 261 |
+
"timeout": "2d",
|
| 262 |
+
"detach": false,
|
| 263 |
+
"tags": []
|
| 264 |
+
},
|
| 265 |
+
"save_checkpoint_to_hub": false,
|
| 266 |
+
"sample_weighting": null,
|
| 267 |
+
"rename_map": {},
|
| 268 |
+
"checkpoint_path": null
|
| 269 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/training_state/optimizer_param_groups.json
ADDED
|
@@ -0,0 +1,249 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"lr": 5.4128967273616625e-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 |
+
]
|
| 123 |
+
},
|
| 124 |
+
{
|
| 125 |
+
"weight_decay": 0.0,
|
| 126 |
+
"lr": 5.4128967273616625e-05,
|
| 127 |
+
"betas": [
|
| 128 |
+
0.9,
|
| 129 |
+
0.999
|
| 130 |
+
],
|
| 131 |
+
"eps": 1e-08,
|
| 132 |
+
"amsgrad": false,
|
| 133 |
+
"maximize": false,
|
| 134 |
+
"foreach": null,
|
| 135 |
+
"capturable": false,
|
| 136 |
+
"differentiable": false,
|
| 137 |
+
"fused": null,
|
| 138 |
+
"decoupled_weight_decay": true,
|
| 139 |
+
"initial_lr": 0.0001,
|
| 140 |
+
"params": [
|
| 141 |
+
103,
|
| 142 |
+
104,
|
| 143 |
+
105,
|
| 144 |
+
106,
|
| 145 |
+
107,
|
| 146 |
+
108,
|
| 147 |
+
109,
|
| 148 |
+
110,
|
| 149 |
+
111,
|
| 150 |
+
112,
|
| 151 |
+
113,
|
| 152 |
+
114,
|
| 153 |
+
115,
|
| 154 |
+
116,
|
| 155 |
+
117,
|
| 156 |
+
118,
|
| 157 |
+
119,
|
| 158 |
+
120,
|
| 159 |
+
121,
|
| 160 |
+
122,
|
| 161 |
+
123,
|
| 162 |
+
124,
|
| 163 |
+
125,
|
| 164 |
+
126,
|
| 165 |
+
127,
|
| 166 |
+
128,
|
| 167 |
+
129,
|
| 168 |
+
130,
|
| 169 |
+
131,
|
| 170 |
+
132,
|
| 171 |
+
133,
|
| 172 |
+
134,
|
| 173 |
+
135,
|
| 174 |
+
136,
|
| 175 |
+
137,
|
| 176 |
+
138,
|
| 177 |
+
139,
|
| 178 |
+
140,
|
| 179 |
+
141,
|
| 180 |
+
142,
|
| 181 |
+
143,
|
| 182 |
+
144,
|
| 183 |
+
145,
|
| 184 |
+
146,
|
| 185 |
+
147,
|
| 186 |
+
148,
|
| 187 |
+
149,
|
| 188 |
+
150,
|
| 189 |
+
151,
|
| 190 |
+
152,
|
| 191 |
+
153,
|
| 192 |
+
154,
|
| 193 |
+
155,
|
| 194 |
+
156,
|
| 195 |
+
157,
|
| 196 |
+
158,
|
| 197 |
+
159,
|
| 198 |
+
160,
|
| 199 |
+
161,
|
| 200 |
+
162,
|
| 201 |
+
163,
|
| 202 |
+
164,
|
| 203 |
+
165,
|
| 204 |
+
166,
|
| 205 |
+
167,
|
| 206 |
+
168,
|
| 207 |
+
169,
|
| 208 |
+
170,
|
| 209 |
+
171,
|
| 210 |
+
172,
|
| 211 |
+
173,
|
| 212 |
+
174,
|
| 213 |
+
175,
|
| 214 |
+
176,
|
| 215 |
+
177,
|
| 216 |
+
178,
|
| 217 |
+
179,
|
| 218 |
+
180,
|
| 219 |
+
181,
|
| 220 |
+
182,
|
| 221 |
+
183,
|
| 222 |
+
184,
|
| 223 |
+
185,
|
| 224 |
+
186,
|
| 225 |
+
187,
|
| 226 |
+
188,
|
| 227 |
+
189,
|
| 228 |
+
190,
|
| 229 |
+
191,
|
| 230 |
+
192,
|
| 231 |
+
193,
|
| 232 |
+
194,
|
| 233 |
+
195,
|
| 234 |
+
196,
|
| 235 |
+
197,
|
| 236 |
+
198,
|
| 237 |
+
199,
|
| 238 |
+
200,
|
| 239 |
+
201,
|
| 240 |
+
202,
|
| 241 |
+
203,
|
| 242 |
+
204,
|
| 243 |
+
205,
|
| 244 |
+
206,
|
| 245 |
+
207,
|
| 246 |
+
208
|
| 247 |
+
]
|
| 248 |
+
}
|
| 249 |
+
]
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000/training_state/training_step.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"step": 30000,
|
| 3 |
+
"num_processes": 1,
|
| 4 |
+
"batch_size": 64
|
| 5 |
+
}
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/wandb/run-20260722_020800-rd8tz18f/files/output.log
ADDED
|
@@ -0,0 +1,226 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO 2026-07-22 02:08:01 db_utils.py:121 Logs will be synced with wandb.
|
| 2 |
+
INFO 2026-07-22 02:08:01 db_utils.py:122 Track this run --> https://wandb.ai/minje227_hyu-hanyang-university/lerobot/runs/rd8tz18f
|
| 3 |
+
INFO 2026-07-22 02:08:01 ot_train.py:298 Creating dataset
|
| 4 |
+
INFO 2026-07-22 02:08:03 ot_train.py:332 Creating policy
|
| 5 |
+
Fetching 27 files: 100%|ββββββββββ| 27/27 [00:00<00:00, 10752.58it/s]
|
| 6 |
+
`torch_dtype` is deprecated! Use `dtype` instead!
|
| 7 |
+
Loading weights: 100%|ββββββββββ| 695/695 [00:00<00:00, 819.98it/s]
|
| 8 |
+
[1mGR00TN17 LOAD REPORT[0m from: /home/ext_minje/.cache/huggingface/hub/models--nvidia--GR00T-N1.7-3B/snapshots/2fc962b973bccdd5d8ce4f67cc63b264d6886495
|
| 9 |
+
Key | Status | |
|
| 10 |
+
--------------------------------------------------------------------+------------+--+-
|
| 11 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.weight | UNEXPECTED | |
|
| 12 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.weight | UNEXPECTED | |
|
| 13 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.weight | UNEXPECTED | |
|
| 14 |
+
action_head.model.transformer_blocks.{8...31}.norm1.linear.bias | UNEXPECTED | |
|
| 15 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_v.bias | UNEXPECTED | |
|
| 16 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.weight | UNEXPECTED | |
|
| 17 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.weight | UNEXPECTED | |
|
| 18 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_q.bias | UNEXPECTED | |
|
| 19 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_out.0.bias | UNEXPECTED | |
|
| 20 |
+
action_head.model.transformer_blocks.{8...31}.attn1.to_k.bias | UNEXPECTED | |
|
| 21 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.weight | UNEXPECTED | |
|
| 22 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.0.proj.bias | UNEXPECTED | |
|
| 23 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.weight | UNEXPECTED | |
|
| 24 |
+
action_head.model.transformer_blocks.{8...31}.ff.net.2.bias | UNEXPECTED | |
|
| 25 |
+
|
| 26 |
+
Notes:
|
| 27 |
+
- UNEXPECTED: can be ignored when loading from different task/architecture; not ok if you expect identical arch.
|
| 28 |
+
INFO 2026-07-22 02:08:09 ot_train.py:405 Creating optimizer and scheduler
|
| 29 |
+
INFO 2026-07-22 02:08:09 ot_train.py:439 Output dir: /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K
|
| 30 |
+
INFO 2026-07-22 02:08:09 ot_train.py:446 cfg.steps=60000 (60K)
|
| 31 |
+
INFO 2026-07-22 02:08:09 ot_train.py:447 dataset.num_frames=218367 (218K)
|
| 32 |
+
INFO 2026-07-22 02:08:09 ot_train.py:448 dataset.num_episodes=4930
|
| 33 |
+
INFO 2026-07-22 02:08:09 ot_train.py:451 Effective batch size: 64 x 1 = 64
|
| 34 |
+
INFO 2026-07-22 02:08:09 ot_train.py:452 num_learnable_params=810087553 (810M)
|
| 35 |
+
INFO 2026-07-22 02:08:09 ot_train.py:453 num_total_params=2333706401 (2B)
|
| 36 |
+
Training: 0%| | 0/60000 [00:00<?, ?step/s]INFO 2026-07-22 02:08:09 ot_train.py:604 Start offline training on a fixed dataset, with effective batch size: 64
|
| 37 |
+
Training: 0%| | 200/60000 [04:17<21:52:19, 1.32s/step]INFO 2026-07-22 02:12:26 ot_train.py:649 step:200 smpl:13K ep:289 epch:0.06 loss:1.332 grdn:0.807 lr:3.3e-06 updt_s:0.708 data_s:0.576 smp/s:50 mem_gb:24.94
|
| 38 |
+
Training: 1%| | 400/60000 [08:27<20:41:10, 1.25s/step]INFO 2026-07-22 02:16:37 ot_train.py:649 step:400 smpl:26K ep:578 epch:0.12 loss:0.780 grdn:2.111 lr:1.0e-05 updt_s:0.678 data_s:0.572 smp/s:51 mem_gb:24.95
|
| 39 |
+
Training: 1%| | 600/60000 [12:35<20:38:48, 1.25s/step]INFO 2026-07-22 02:20:45 ot_train.py:649 step:600 smpl:38K ep:867 epch:0.18 loss:0.382 grdn:2.395 lr:1.7e-05 updt_s:0.671 data_s:0.568 smp/s:52 mem_gb:24.95
|
| 40 |
+
Training: 1%|β | 800/60000 [16:44<20:18:11, 1.23s/step]INFO 2026-07-22 02:24:54 ot_train.py:649 step:800 smpl:51K ep:1K epch:0.23 loss:0.299 grdn:2.202 lr:2.3e-05 updt_s:0.673 data_s:0.569 smp/s:52 mem_gb:24.95
|
| 41 |
+
Training: 2%|β | 1000/60000 [20:54<20:24:18, 1.25s/step]INFO 2026-07-22 02:29:03 ot_train.py:649 step:1K smpl:64K ep:1K epch:0.29 loss:0.247 grdn:1.973 lr:3.0e-05 updt_s:0.679 data_s:0.564 smp/s:51 mem_gb:24.95
|
| 42 |
+
Training: 2%|β | 1200/60000 [25:02<20:04:30, 1.23s/step]INFO 2026-07-22 02:33:11 ot_train.py:649 step:1K smpl:77K ep:2K epch:0.35 loss:0.217 grdn:1.622 lr:3.7e-05 updt_s:0.673 data_s:0.565 smp/s:52 mem_gb:24.95
|
| 43 |
+
Training: 2%|β | 1400/60000 [29:11<19:47:37, 1.22s/step]INFO 2026-07-22 02:37:20 ot_train.py:649 step:1K smpl:90K ep:2K epch:0.41 loss:0.201 grdn:1.374 lr:4.3e-05 updt_s:0.676 data_s:0.565 smp/s:52 mem_gb:24.95
|
| 44 |
+
Training: 3%|β | 1600/60000 [33:19<19:51:53, 1.22s/step]INFO 2026-07-22 02:41:28 ot_train.py:649 step:2K smpl:102K ep:2K epch:0.47 loss:0.189 grdn:1.126 lr:5.0e-05 updt_s:0.675 data_s:0.562 smp/s:52 mem_gb:24.95
|
| 45 |
+
Training: 3%|β | 1800/60000 [37:28<20:34:16, 1.27s/step]INFO 2026-07-22 02:45:37 ot_train.py:649 step:2K smpl:115K ep:3K epch:0.53 loss:0.176 grdn:0.993 lr:5.7e-05 updt_s:0.677 data_s:0.566 smp/s:51 mem_gb:24.95
|
| 46 |
+
Training: 3%|β | 2000/60000 [41:45<20:35:14, 1.28s/step]INFO 2026-07-22 02:49:54 ot_train.py:649 step:2K smpl:128K ep:3K epch:0.59 loss:0.163 grdn:0.867 lr:6.3e-05 updt_s:0.691 data_s:0.591 smp/s:50 mem_gb:24.95
|
| 47 |
+
Training: 4%|β | 2200/60000 [46:06<20:26:04, 1.27s/step]INFO 2026-07-22 02:54:16 ot_train.py:649 step:2K smpl:141K ep:3K epch:0.64 loss:0.158 grdn:0.802 lr:7.0e-05 updt_s:0.707 data_s:0.598 smp/s:49 mem_gb:24.95
|
| 48 |
+
Training: 4%|β | 2400/60000 [50:23<19:47:57, 1.24s/step]INFO 2026-07-22 02:58:33 ot_train.py:649 step:2K smpl:154K ep:3K epch:0.70 loss:0.152 grdn:0.755 lr:7.7e-05 updt_s:0.727 data_s:0.555 smp/s:50 mem_gb:24.93
|
| 49 |
+
Training: 4%|β | 2600/60000 [54:31<19:50:37, 1.24s/step]INFO 2026-07-22 03:02:41 ot_train.py:649 step:3K smpl:166K ep:4K epch:0.76 loss:0.148 grdn:0.692 lr:8.3e-05 updt_s:0.678 data_s:0.559 smp/s:52 mem_gb:24.95
|
| 50 |
+
Training: 5%|β | 2800/60000 [58:49<25:39:55, 1.62s/step]INFO 2026-07-22 03:06:59 ot_train.py:649 step:3K smpl:179K ep:4K epch:0.82 loss:0.144 grdn:0.671 lr:9.0e-05 updt_s:0.719 data_s:0.569 smp/s:50 mem_gb:24.95
|
| 51 |
+
Training: 5%|β | 3000/60000 [1:03:01<19:13:40, 1.21s/step]INFO 2026-07-22 03:11:10 ot_train.py:649 step:3K smpl:192K ep:4K epch:0.88 loss:0.139 grdn:0.623 lr:9.7e-05 updt_s:0.687 data_s:0.569 smp/s:51 mem_gb:24.95
|
| 52 |
+
Training: 5%|β | 3200/60000 [1:07:15<20:52:08, 1.32s/step]INFO 2026-07-22 03:15:24 ot_train.py:649 step:3K smpl:205K ep:5K epch:0.94 loss:0.138 grdn:0.584 lr:1.0e-04 updt_s:0.703 data_s:0.563 smp/s:51 mem_gb:24.95
|
| 53 |
+
Training: 6%|β | 3400/60000 [1:11:23<19:46:03, 1.26s/step]INFO 2026-07-22 03:19:33 ot_train.py:649 step:3K smpl:218K ep:5K epch:1.00 loss:0.134 grdn:0.559 lr:1.0e-04 updt_s:0.678 data_s:0.562 smp/s:52 mem_gb:24.95
|
| 54 |
+
Training: 6%|β | 3600/60000 [1:16:32<19:29:32, 1.24s/step]INFO 2026-07-22 03:24:41 ot_train.py:649 step:4K smpl:230K ep:5K epch:1.06 loss:0.127 grdn:0.532 lr:1.0e-04 updt_s:0.943 data_s:0.597 smp/s:42 mem_gb:24.95
|
| 55 |
+
Training: 6%|β | 3800/60000 [1:20:38<19:13:26, 1.23s/step]INFO 2026-07-22 03:28:47 ot_train.py:649 step:4K smpl:243K ep:5K epch:1.11 loss:0.122 grdn:0.503 lr:1.0e-04 updt_s:0.678 data_s:0.551 smp/s:52 mem_gb:24.95
|
| 56 |
+
Training: 7%|β | 4000/60000 [1:25:04<19:13:53, 1.24s/step]INFO 2026-07-22 03:33:13 ot_train.py:649 step:4K smpl:256K ep:6K epch:1.17 loss:0.120 grdn:0.488 lr:1.0e-04 updt_s:0.766 data_s:0.561 smp/s:48 mem_gb:24.95
|
| 57 |
+
Training: 7%|β | 4200/60000 [1:29:13<18:52:47, 1.22s/step]INFO 2026-07-22 03:37:23 ot_train.py:649 step:4K smpl:269K ep:6K epch:1.23 loss:0.119 grdn:0.487 lr:1.0e-04 updt_s:0.677 data_s:0.567 smp/s:51 mem_gb:24.95
|
| 58 |
+
Training: 7%|β | 4400/60000 [1:33:22<18:56:28, 1.23s/step]INFO 2026-07-22 03:41:32 ot_train.py:649 step:4K smpl:282K ep:6K epch:1.29 loss:0.115 grdn:0.469 lr:1.0e-04 updt_s:0.677 data_s:0.565 smp/s:51 mem_gb:24.95
|
| 59 |
+
Training: 8%|β | 4600/60000 [1:37:32<19:57:23, 1.30s/step]INFO 2026-07-22 03:45:42 ot_train.py:649 step:5K smpl:294K ep:7K epch:1.35 loss:0.116 grdn:0.468 lr:1.0e-04 updt_s:0.679 data_s:0.567 smp/s:51 mem_gb:24.93
|
| 60 |
+
Training: 8%|β | 4800/60000 [1:41:41<19:09:37, 1.25s/step]INFO 2026-07-22 03:49:51 ot_train.py:649 step:5K smpl:307K ep:7K epch:1.41 loss:0.109 grdn:0.434 lr:1.0e-04 updt_s:0.680 data_s:0.564 smp/s:51 mem_gb:24.95
|
| 61 |
+
Training: 8%|β | 5000/60000 [1:45:51<18:45:26, 1.23s/step]INFO 2026-07-22 03:54:01 ot_train.py:649 step:5K smpl:320K ep:7K epch:1.47 loss:0.110 grdn:0.447 lr:1.0e-04 updt_s:0.678 data_s:0.568 smp/s:51 mem_gb:24.95
|
| 62 |
+
Training: 9%|β | 5200/60000 [1:50:01<19:14:48, 1.26s/step]INFO 2026-07-22 03:58:11 ot_train.py:649 step:5K smpl:333K ep:8K epch:1.52 loss:0.109 grdn:0.432 lr:1.0e-04 updt_s:0.681 data_s:0.567 smp/s:51 mem_gb:24.95
|
| 63 |
+
Training: 9%|β | 5400/60000 [1:54:12<18:38:16, 1.23s/step]INFO 2026-07-22 04:02:21 ot_train.py:649 step:5K smpl:346K ep:8K epch:1.58 loss:0.110 grdn:0.446 lr:1.0e-04 updt_s:0.683 data_s:0.569 smp/s:51 mem_gb:24.95
|
| 64 |
+
Training: 9%|β | 5600/60000 [1:58:21<18:29:01, 1.22s/step]INFO 2026-07-22 04:06:31 ot_train.py:649 step:6K smpl:358K ep:8K epch:1.64 loss:0.107 grdn:0.434 lr:1.0e-04 updt_s:0.673 data_s:0.570 smp/s:51 mem_gb:24.95
|
| 65 |
+
Training: 10%|β | 5800/60000 [2:02:30<18:22:02, 1.22s/step]INFO 2026-07-22 04:10:40 ot_train.py:649 step:6K smpl:371K ep:8K epch:1.70 loss:0.104 grdn:0.410 lr:9.9e-05 updt_s:0.673 data_s:0.569 smp/s:52 mem_gb:24.95
|
| 66 |
+
Training: 10%|β | 6000/60000 [2:06:41<18:19:38, 1.22s/step]INFO 2026-07-22 04:14:50 ot_train.py:649 step:6K smpl:384K ep:9K epch:1.76 loss:0.104 grdn:0.410 lr:9.9e-05 updt_s:0.684 data_s:0.568 smp/s:51 mem_gb:24.95
|
| 67 |
+
Training: 10%|β | 6200/60000 [2:10:49<18:27:13, 1.23s/step]INFO 2026-07-22 04:18:58 ot_train.py:649 step:6K smpl:397K ep:9K epch:1.82 loss:0.104 grdn:0.411 lr:9.9e-05 updt_s:0.674 data_s:0.564 smp/s:52 mem_gb:24.95
|
| 68 |
+
Training: 11%|β | 6400/60000 [2:15:03<18:35:41, 1.25s/step]INFO 2026-07-22 04:23:13 ot_train.py:649 step:6K smpl:410K ep:9K epch:1.88 loss:0.100 grdn:0.399 lr:9.9e-05 updt_s:0.704 data_s:0.564 smp/s:50 mem_gb:24.95
|
| 69 |
+
Training: 11%|β | 6600/60000 [2:19:25<19:19:48, 1.30s/step]INFO 2026-07-22 04:27:34 ot_train.py:649 step:7K smpl:422K ep:10K epch:1.93 loss:0.101 grdn:0.397 lr:9.9e-05 updt_s:0.702 data_s:0.604 smp/s:49 mem_gb:24.95
|
| 70 |
+
Training: 11%|ββ | 6800/60000 [2:23:43<18:37:35, 1.26s/step]INFO 2026-07-22 04:31:53 ot_train.py:649 step:7K smpl:435K ep:10K epch:1.99 loss:0.097 grdn:0.387 lr:9.9e-05 updt_s:0.695 data_s:0.596 smp/s:50 mem_gb:24.93
|
| 71 |
+
Training: 12%|ββ | 7000/60000 [2:28:00<18:22:47, 1.25s/step]INFO 2026-07-22 04:36:09 ot_train.py:649 step:7K smpl:448K ep:10K epch:2.05 loss:0.100 grdn:0.398 lr:9.9e-05 updt_s:0.691 data_s:0.587 smp/s:50 mem_gb:24.95
|
| 72 |
+
Training: 12%|ββ | 7200/60000 [2:32:36<21:01:28, 1.43s/step]INFO 2026-07-22 04:40:45 ot_train.py:649 step:7K smpl:461K ep:10K epch:2.11 loss:0.098 grdn:0.403 lr:9.9e-05 updt_s:0.727 data_s:0.650 smp/s:46 mem_gb:24.95
|
| 73 |
+
Training: 12%|ββ | 7400/60000 [2:37:49<24:52:45, 1.70s/step]INFO 2026-07-22 04:45:58 ot_train.py:649 step:7K smpl:474K ep:11K epch:2.17 loss:0.096 grdn:0.384 lr:9.9e-05 updt_s:0.814 data_s:0.748 smp/s:41 mem_gb:24.95
|
| 74 |
+
Training: 13%|ββ | 7600/60000 [2:43:00<20:27:40, 1.41s/step]INFO 2026-07-22 04:51:10 ot_train.py:649 step:8K smpl:486K ep:11K epch:2.23 loss:0.098 grdn:0.402 lr:9.8e-05 updt_s:0.802 data_s:0.750 smp/s:41 mem_gb:24.95
|
| 75 |
+
Training: 13%|ββ | 7800/60000 [2:48:12<21:21:18, 1.47s/step]INFO 2026-07-22 04:56:22 ot_train.py:649 step:8K smpl:499K ep:11K epch:2.29 loss:0.096 grdn:0.384 lr:9.8e-05 updt_s:0.775 data_s:0.782 smp/s:41 mem_gb:24.95
|
| 76 |
+
Training: 13%|ββ | 8000/60000 [2:54:02<22:51:48, 1.58s/step]INFO 2026-07-22 05:02:12 ot_train.py:649 step:8K smpl:512K ep:12K epch:2.34 loss:0.094 grdn:0.377 lr:9.8e-05 updt_s:0.732 data_s:1.016 smp/s:37 mem_gb:24.95
|
| 77 |
+
Training: 14%|ββ | 8200/60000 [3:00:23<26:21:50, 1.83s/step]INFO 2026-07-22 05:08:32 ot_train.py:649 step:8K smpl:525K ep:12K epch:2.40 loss:0.095 grdn:0.378 lr:9.8e-05 updt_s:0.770 data_s:1.128 smp/s:34 mem_gb:24.95
|
| 78 |
+
Training: 14%|ββ | 8400/60000 [3:06:31<23:23:09, 1.63s/step]INFO 2026-07-22 05:14:40 ot_train.py:649 step:8K smpl:538K ep:12K epch:2.46 loss:0.093 grdn:0.372 lr:9.8e-05 updt_s:0.782 data_s:1.056 smp/s:35 mem_gb:24.95
|
| 79 |
+
Training: 14%|ββ | 8600/60000 [3:12:47<29:41:52, 2.08s/step]INFO 2026-07-22 05:20:57 ot_train.py:649 step:9K smpl:550K ep:12K epch:2.52 loss:0.094 grdn:0.393 lr:9.8e-05 updt_s:0.774 data_s:1.105 smp/s:34 mem_gb:24.95
|
| 80 |
+
Training: 15%|ββ | 8800/60000 [3:17:24<17:59:12, 1.26s/step]INFO 2026-07-22 05:25:34 ot_train.py:649 step:9K smpl:563K ep:13K epch:2.58 loss:0.094 grdn:0.374 lr:9.8e-05 updt_s:0.696 data_s:0.684 smp/s:46 mem_gb:24.95
|
| 81 |
+
Training: 15%|ββ | 9000/60000 [3:21:36<20:07:38, 1.42s/step]INFO 2026-07-22 05:29:46 ot_train.py:649 step:9K smpl:576K ep:13K epch:2.64 loss:0.090 grdn:0.362 lr:9.7e-05 updt_s:0.710 data_s:0.547 smp/s:51 mem_gb:24.95
|
| 82 |
+
Training: 15%|ββ | 9200/60000 [3:26:40<21:29:33, 1.52s/step]INFO 2026-07-22 05:34:49 ot_train.py:649 step:9K smpl:589K ep:13K epch:2.70 loss:0.089 grdn:0.359 lr:9.7e-05 updt_s:0.774 data_s:0.742 smp/s:42 mem_gb:24.93
|
| 83 |
+
Training: 16%|ββ | 9400/60000 [3:31:46<21:38:04, 1.54s/step]INFO 2026-07-22 05:39:55 ot_train.py:649 step:9K smpl:602K ep:14K epch:2.75 loss:0.088 grdn:0.362 lr:9.7e-05 updt_s:0.768 data_s:0.758 smp/s:42 mem_gb:24.95
|
| 84 |
+
Training: 16%|ββ | 9600/60000 [3:36:52<22:04:05, 1.58s/step]INFO 2026-07-22 05:45:01 ot_train.py:649 step:10K smpl:614K ep:14K epch:2.81 loss:0.089 grdn:0.365 lr:9.7e-05 updt_s:0.766 data_s:0.760 smp/s:42 mem_gb:24.95
|
| 85 |
+
Training: 16%|ββ | 9800/60000 [3:42:38<23:40:20, 1.70s/step]INFO 2026-07-22 05:50:47 ot_train.py:649 step:10K smpl:627K ep:14K epch:2.87 loss:0.090 grdn:0.371 lr:9.7e-05 updt_s:0.772 data_s:0.956 smp/s:37 mem_gb:24.95
|
| 86 |
+
Training: 17%|ββ | 10000/60000 [3:47:59<21:54:35, 1.58s/step]INFO 2026-07-22 05:56:09 ot_train.py:649 step:10K smpl:640K ep:14K epch:2.93 loss:0.088 grdn:0.359 lr:9.6e-05 updt_s:0.771 data_s:0.834 smp/s:40 mem_gb:24.95
|
| 87 |
+
INFO 2026-07-22 05:56:09 ot_train.py:694 Checkpoint policy after step 10000
|
| 88 |
+
INFO 2026-07-22 05:56:40 in_yaml.py:1270 Saved config, action-space, prompt, and phase provenance in /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K and /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/010000
|
| 89 |
+
Training: 17%|ββ | 10200/60000 [3:55:17<20:43:28, 1.50s/step]INFO 2026-07-22 06:03:27 ot_train.py:649 step:10K smpl:653K ep:15K epch:2.99 loss:0.087 grdn:0.366 lr:9.6e-05 updt_s:0.759 data_s:1.275 smp/s:31 mem_gb:24.95
|
| 90 |
+
Training: 17%|ββ | 10400/60000 [4:01:04<25:42:44, 1.87s/step]INFO 2026-07-22 06:09:13 ot_train.py:649 step:10K smpl:666K ep:15K epch:3.05 loss:0.085 grdn:0.360 lr:9.6e-05 updt_s:0.779 data_s:0.948 smp/s:37 mem_gb:24.95
|
| 91 |
+
Training: 18%|ββ | 10600/60000 [4:06:31<20:51:22, 1.52s/step]INFO 2026-07-22 06:14:41 ot_train.py:649 step:11K smpl:678K ep:15K epch:3.11 loss:0.086 grdn:0.353 lr:9.6e-05 updt_s:0.789 data_s:0.847 smp/s:39 mem_gb:24.95
|
| 92 |
+
Training: 18%|ββ | 10800/60000 [4:11:49<21:44:09, 1.59s/step]INFO 2026-07-22 06:19:58 ot_train.py:649 step:11K smpl:691K ep:16K epch:3.17 loss:0.083 grdn:0.357 lr:9.6e-05 updt_s:0.818 data_s:0.764 smp/s:40 mem_gb:24.95
|
| 93 |
+
Training: 18%|ββ | 11000/60000 [4:17:08<21:31:46, 1.58s/step]INFO 2026-07-22 06:25:18 ot_train.py:649 step:11K smpl:704K ep:16K epch:3.22 loss:0.082 grdn:0.359 lr:9.5e-05 updt_s:0.807 data_s:0.789 smp/s:40 mem_gb:24.95
|
| 94 |
+
Training: 19%|ββ | 11200/60000 [4:23:04<23:55:50, 1.77s/step]INFO 2026-07-22 06:31:14 ot_train.py:649 step:11K smpl:717K ep:16K epch:3.28 loss:0.085 grdn:0.356 lr:9.5e-05 updt_s:0.815 data_s:0.961 smp/s:36 mem_gb:24.95
|
| 95 |
+
Training: 19%|ββ | 11400/60000 [4:28:53<31:03:55, 2.30s/step]INFO 2026-07-22 06:37:02 ot_train.py:649 step:11K smpl:730K ep:16K epch:3.34 loss:0.083 grdn:0.351 lr:9.5e-05 updt_s:0.779 data_s:0.960 smp/s:37 mem_gb:24.93
|
| 96 |
+
Training: 19%|ββ | 11600/60000 [4:35:03<27:23:43, 2.04s/step]INFO 2026-07-22 06:43:13 ot_train.py:649 step:12K smpl:742K ep:17K epch:3.40 loss:0.081 grdn:0.359 lr:9.5e-05 updt_s:0.873 data_s:0.975 smp/s:35 mem_gb:24.95
|
| 97 |
+
Training: 20%|ββ | 11800/60000 [4:40:38<23:06:30, 1.73s/step]INFO 2026-07-22 06:48:48 ot_train.py:649 step:12K smpl:755K ep:17K epch:3.46 loss:0.082 grdn:0.350 lr:9.4e-05 updt_s:0.837 data_s:0.835 smp/s:38 mem_gb:24.95
|
| 98 |
+
Training: 20%|ββ | 12000/60000 [4:46:10<20:35:18, 1.54s/step]INFO 2026-07-22 06:54:19 ot_train.py:649 step:12K smpl:768K ep:17K epch:3.52 loss:0.080 grdn:0.348 lr:9.4e-05 updt_s:0.811 data_s:0.843 smp/s:39 mem_gb:24.95
|
| 99 |
+
Training: 20%|ββ | 12200/60000 [4:51:30<19:53:32, 1.50s/step]INFO 2026-07-22 06:59:40 ot_train.py:649 step:12K smpl:781K ep:18K epch:3.58 loss:0.079 grdn:0.347 lr:9.4e-05 updt_s:0.823 data_s:0.775 smp/s:40 mem_gb:24.95
|
| 100 |
+
Training: 21%|ββ | 12400/60000 [4:57:39<27:17:04, 2.06s/step]INFO 2026-07-22 07:05:48 ot_train.py:649 step:12K smpl:794K ep:18K epch:3.63 loss:0.078 grdn:0.355 lr:9.4e-05 updt_s:0.793 data_s:1.046 smp/s:35 mem_gb:24.95
|
| 101 |
+
Training: 21%|ββ | 12600/60000 [5:04:05<21:04:48, 1.60s/step]INFO 2026-07-22 07:12:15 ot_train.py:649 step:13K smpl:806K ep:18K epch:3.69 loss:0.080 grdn:0.346 lr:9.3e-05 updt_s:0.818 data_s:1.111 smp/s:33 mem_gb:24.95
|
| 102 |
+
Training: 21%|βββ | 12800/60000 [5:09:15<16:59:54, 1.30s/step]INFO 2026-07-22 07:17:24 ot_train.py:649 step:13K smpl:819K ep:18K epch:3.75 loss:0.080 grdn:0.359 lr:9.3e-05 updt_s:0.794 data_s:0.750 smp/s:41 mem_gb:24.95
|
| 103 |
+
Training: 22%|βββ | 13000/60000 [5:13:43<17:26:31, 1.34s/step]INFO 2026-07-22 07:21:53 ot_train.py:649 step:13K smpl:832K ep:19K epch:3.81 loss:0.079 grdn:0.349 lr:9.3e-05 updt_s:0.717 data_s:0.622 smp/s:48 mem_gb:24.95
|
| 104 |
+
Training: 22%|βββ | 13200/60000 [5:18:23<20:45:55, 1.60s/step]INFO 2026-07-22 07:26:33 ot_train.py:649 step:13K smpl:845K ep:19K epch:3.87 loss:0.079 grdn:0.344 lr:9.2e-05 updt_s:0.751 data_s:0.646 smp/s:46 mem_gb:24.95
|
| 105 |
+
Training: 22%|βββ | 13400/60000 [5:23:17<20:27:40, 1.58s/step]INFO 2026-07-22 07:31:27 ot_train.py:649 step:13K smpl:858K ep:19K epch:3.93 loss:0.079 grdn:0.344 lr:9.2e-05 updt_s:0.760 data_s:0.708 smp/s:44 mem_gb:24.95
|
| 106 |
+
Training: 23%|βββ | 13600/60000 [5:27:48<18:23:46, 1.43s/step]INFO 2026-07-22 07:35:57 ot_train.py:649 step:14K smpl:870K ep:20K epch:3.99 loss:0.077 grdn:0.340 lr:9.2e-05 updt_s:0.712 data_s:0.637 smp/s:47 mem_gb:24.93
|
| 107 |
+
Training: 23%|βββ | 13800/60000 [5:32:06<16:02:02, 1.25s/step]INFO 2026-07-22 07:40:15 ot_train.py:649 step:14K smpl:883K ep:20K epch:4.04 loss:0.073 grdn:0.334 lr:9.2e-05 updt_s:0.695 data_s:0.592 smp/s:50 mem_gb:24.95
|
| 108 |
+
Training: 23%|βββ | 14000/60000 [5:36:28<16:08:52, 1.26s/step]INFO 2026-07-22 07:44:38 ot_train.py:649 step:14K smpl:896K ep:20K epch:4.10 loss:0.075 grdn:0.354 lr:9.1e-05 updt_s:0.730 data_s:0.580 smp/s:49 mem_gb:24.95
|
| 109 |
+
Training: 24%|βββ | 14200/60000 [5:40:49<15:40:17, 1.23s/step]INFO 2026-07-22 07:48:59 ot_train.py:649 step:14K smpl:909K ep:21K epch:4.16 loss:0.071 grdn:0.328 lr:9.1e-05 updt_s:0.709 data_s:0.592 smp/s:49 mem_gb:24.95
|
| 110 |
+
Training: 24%|βββ | 14400/60000 [5:45:05<16:17:10, 1.29s/step]INFO 2026-07-22 07:53:15 ot_train.py:649 step:14K smpl:922K ep:21K epch:4.22 loss:0.074 grdn:0.335 lr:9.1e-05 updt_s:0.693 data_s:0.585 smp/s:50 mem_gb:24.95
|
| 111 |
+
Training: 24%|βββ | 14600/60000 [5:49:20<19:51:38, 1.57s/step]INFO 2026-07-22 07:57:30 ot_train.py:649 step:15K smpl:934K ep:21K epch:4.28 loss:0.074 grdn:0.345 lr:9.0e-05 updt_s:0.687 data_s:0.586 smp/s:50 mem_gb:24.95
|
| 112 |
+
Training: 25%|βββ | 14800/60000 [5:53:38<16:30:54, 1.32s/step]INFO 2026-07-22 08:01:47 ot_train.py:649 step:15K smpl:947K ep:21K epch:4.34 loss:0.076 grdn:0.352 lr:9.0e-05 updt_s:0.695 data_s:0.589 smp/s:50 mem_gb:24.95
|
| 113 |
+
Training: 25%|βββ | 15000/60000 [5:57:55<15:43:22, 1.26s/step]INFO 2026-07-22 08:06:04 ot_train.py:649 step:15K smpl:960K ep:22K epch:4.40 loss:0.072 grdn:0.334 lr:9.0e-05 updt_s:0.694 data_s:0.587 smp/s:50 mem_gb:24.95
|
| 114 |
+
Training: 25%|βββ | 15200/60000 [6:02:14<15:24:03, 1.24s/step]INFO 2026-07-22 08:10:24 ot_train.py:649 step:15K smpl:973K ep:22K epch:4.45 loss:0.075 grdn:0.344 lr:8.9e-05 updt_s:0.694 data_s:0.600 smp/s:49 mem_gb:24.95
|
| 115 |
+
Training: 26%|βββ | 15400/60000 [6:06:36<28:27:09, 2.30s/step]INFO 2026-07-22 08:14:45 ot_train.py:649 step:15K smpl:986K ep:22K epch:4.51 loss:0.073 grdn:0.339 lr:8.9e-05 updt_s:0.687 data_s:0.619 smp/s:49 mem_gb:24.95
|
| 116 |
+
Training: 26%|βββ | 15600/60000 [6:13:19<25:43:04, 2.09s/step]INFO 2026-07-22 08:21:29 ot_train.py:649 step:16K smpl:998K ep:23K epch:4.57 loss:0.071 grdn:0.326 lr:8.9e-05 updt_s:0.796 data_s:1.218 smp/s:32 mem_gb:24.95
|
| 117 |
+
Training: 26%|βββ | 15800/60000 [6:18:40<24:37:17, 2.01s/step]INFO 2026-07-22 08:26:49 ot_train.py:649 step:16K smpl:1M ep:23K epch:4.63 loss:0.070 grdn:0.328 lr:8.8e-05 updt_s:0.737 data_s:0.861 smp/s:40 mem_gb:24.93
|
| 118 |
+
Training: 27%|βββ | 16000/60000 [6:23:52<18:10:10, 1.49s/step]INFO 2026-07-22 08:32:02 ot_train.py:649 step:16K smpl:1M ep:23K epch:4.69 loss:0.070 grdn:0.341 lr:8.8e-05 updt_s:0.722 data_s:0.839 smp/s:41 mem_gb:24.95
|
| 119 |
+
Training: 27%|βββ | 16200/60000 [6:29:12<19:27:13, 1.60s/step]INFO 2026-07-22 08:37:21 ot_train.py:649 step:16K smpl:1M ep:23K epch:4.75 loss:0.069 grdn:0.328 lr:8.8e-05 updt_s:0.725 data_s:0.869 smp/s:40 mem_gb:24.95
|
| 120 |
+
Training: 27%|βββ | 16400/60000 [6:35:23<19:14:49, 1.59s/step]INFO 2026-07-22 08:43:32 ot_train.py:649 step:16K smpl:1M ep:24K epch:4.81 loss:0.070 grdn:0.341 lr:8.7e-05 updt_s:0.768 data_s:1.085 smp/s:35 mem_gb:24.95
|
| 121 |
+
Training: 28%|βββ | 16600/60000 [6:40:55<24:42:53, 2.05s/step]INFO 2026-07-22 08:49:05 ot_train.py:649 step:17K smpl:1M ep:24K epch:4.87 loss:0.069 grdn:0.338 lr:8.7e-05 updt_s:0.748 data_s:0.912 smp/s:39 mem_gb:24.95
|
| 122 |
+
Training: 28%|βββ | 16800/60000 [6:47:32<35:55:49, 2.99s/step]INFO 2026-07-22 08:55:41 ot_train.py:649 step:17K smpl:1M ep:24K epch:4.92 loss:0.067 grdn:0.324 lr:8.6e-05 updt_s:0.771 data_s:1.207 smp/s:32 mem_gb:24.95
|
| 123 |
+
Training: 28%|βββ | 17000/60000 [6:52:24<16:35:32, 1.39s/step]INFO 2026-07-22 09:00:34 ot_train.py:649 step:17K smpl:1M ep:25K epch:4.98 loss:0.068 grdn:0.339 lr:8.6e-05 updt_s:0.698 data_s:0.762 smp/s:44 mem_gb:24.95
|
| 124 |
+
Training: 29%|βββ | 17200/60000 [6:58:09<23:17:26, 1.96s/step]INFO 2026-07-22 09:06:19 ot_train.py:649 step:17K smpl:1M ep:25K epch:5.04 loss:0.067 grdn:0.343 lr:8.6e-05 updt_s:0.772 data_s:0.951 smp/s:37 mem_gb:24.95
|
| 125 |
+
Training: 29%|βββ | 17400/60000 [7:04:07<25:18:32, 2.14s/step]INFO 2026-07-22 09:12:17 ot_train.py:649 step:17K smpl:1M ep:25K epch:5.10 loss:0.068 grdn:0.332 lr:8.5e-05 updt_s:0.775 data_s:1.010 smp/s:36 mem_gb:24.95
|
| 126 |
+
Training: 29%|βββ | 17600/60000 [7:09:12<16:39:30, 1.41s/step]INFO 2026-07-22 09:17:21 ot_train.py:649 step:18K smpl:1M ep:25K epch:5.16 loss:0.069 grdn:0.345 lr:8.5e-05 updt_s:0.700 data_s:0.820 smp/s:42 mem_gb:24.95
|
| 127 |
+
Training: 30%|βββ | 17800/60000 [7:14:05<22:05:15, 1.88s/step]INFO 2026-07-22 09:22:15 ot_train.py:649 step:18K smpl:1M ep:26K epch:5.22 loss:0.068 grdn:0.341 lr:8.4e-05 updt_s:0.689 data_s:0.776 smp/s:44 mem_gb:24.95
|
| 128 |
+
Training: 30%|βββ | 18000/60000 [7:20:06<15:41:03, 1.34s/step]INFO 2026-07-22 09:28:16 ot_train.py:649 step:18K smpl:1M ep:26K epch:5.28 loss:0.067 grdn:0.334 lr:8.4e-05 updt_s:0.736 data_s:1.067 smp/s:35 mem_gb:24.95
|
| 129 |
+
Training: 30%|βββ | 18200/60000 [7:24:57<17:11:00, 1.48s/step]INFO 2026-07-22 09:33:07 ot_train.py:649 step:18K smpl:1M ep:26K epch:5.33 loss:0.064 grdn:0.332 lr:8.4e-05 updt_s:0.717 data_s:0.734 smp/s:44 mem_gb:24.93
|
| 130 |
+
Training: 31%|βββ | 18400/60000 [7:29:39<17:33:33, 1.52s/step]INFO 2026-07-22 09:37:49 ot_train.py:649 step:18K smpl:1M ep:27K epch:5.39 loss:0.063 grdn:0.339 lr:8.3e-05 updt_s:0.705 data_s:0.703 smp/s:45 mem_gb:24.95
|
| 131 |
+
Training: 31%|βββ | 18600/60000 [7:34:13<14:19:24, 1.25s/step]INFO 2026-07-22 09:42:22 ot_train.py:649 step:19K smpl:1M ep:27K epch:5.45 loss:0.062 grdn:0.321 lr:8.3e-05 updt_s:0.679 data_s:0.685 smp/s:47 mem_gb:24.95
|
| 132 |
+
Training: 31%|ββββ | 18800/60000 [7:39:18<20:42:37, 1.81s/step]INFO 2026-07-22 09:47:27 ot_train.py:649 step:19K smpl:1M ep:27K epch:5.51 loss:0.064 grdn:0.352 lr:8.2e-05 updt_s:0.711 data_s:0.810 smp/s:42 mem_gb:24.95
|
| 133 |
+
Training: 32%|ββββ | 19000/60000 [7:44:04<14:09:10, 1.24s/step]INFO 2026-07-22 09:52:13 ot_train.py:649 step:19K smpl:1M ep:27K epch:5.57 loss:0.066 grdn:0.347 lr:8.2e-05 updt_s:0.665 data_s:0.764 smp/s:45 mem_gb:24.95
|
| 134 |
+
Training: 32%|ββββ | 19200/60000 [7:48:31<15:09:29, 1.34s/step]INFO 2026-07-22 09:56:40 ot_train.py:649 step:19K smpl:1M ep:28K epch:5.63 loss:0.063 grdn:0.340 lr:8.2e-05 updt_s:0.656 data_s:0.676 smp/s:48 mem_gb:24.95
|
| 135 |
+
Training: 32%|ββββ | 19400/60000 [7:53:48<25:18:20, 2.24s/step]INFO 2026-07-22 10:01:57 ot_train.py:649 step:19K smpl:1M ep:28K epch:5.69 loss:0.061 grdn:0.333 lr:8.1e-05 updt_s:0.705 data_s:0.878 smp/s:40 mem_gb:24.95
|
| 136 |
+
Training: 33%|ββββ | 19600/60000 [7:58:34<14:01:16, 1.25s/step]INFO 2026-07-22 10:06:43 ot_train.py:649 step:20K smpl:1M ep:28K epch:5.74 loss:0.063 grdn:0.336 lr:8.1e-05 updt_s:0.692 data_s:0.736 smp/s:45 mem_gb:24.95
|
| 137 |
+
Training: 33%|ββββ | 19800/60000 [8:02:55<14:58:11, 1.34s/step]INFO 2026-07-22 10:11:05 ot_train.py:649 step:20K smpl:1M ep:29K epch:5.80 loss:0.062 grdn:0.338 lr:8.0e-05 updt_s:0.672 data_s:0.633 smp/s:49 mem_gb:24.95
|
| 138 |
+
Training: 33%|ββββ | 20000/60000 [8:07:14<13:57:34, 1.26s/step]INFO 2026-07-22 10:15:23 ot_train.py:649 step:20K smpl:1M ep:29K epch:5.86 loss:0.063 grdn:0.336 lr:8.0e-05 updt_s:0.657 data_s:0.632 smp/s:50 mem_gb:24.95
|
| 139 |
+
Training: 34%|ββββ | 20200/60000 [8:11:37<16:52:51, 1.53s/step]INFO 2026-07-22 10:19:46 ot_train.py:649 step:20K smpl:1M ep:29K epch:5.92 loss:0.062 grdn:0.336 lr:7.9e-05 updt_s:0.668 data_s:0.644 smp/s:49 mem_gb:24.95
|
| 140 |
+
Training: 34%|ββββ | 20400/60000 [8:17:02<22:25:14, 2.04s/step]INFO 2026-07-22 10:25:11 ot_train.py:649 step:20K smpl:1M ep:29K epch:5.98 loss:0.060 grdn:0.338 lr:7.9e-05 updt_s:0.830 data_s:0.790 smp/s:39 mem_gb:24.93
|
| 141 |
+
Training: 34%|ββββ | 20600/60000 [8:23:11<16:36:52, 1.52s/step]INFO 2026-07-22 10:31:21 ot_train.py:649 step:21K smpl:1M ep:30K epch:6.04 loss:0.059 grdn:0.336 lr:7.8e-05 updt_s:0.945 data_s:0.899 smp/s:35 mem_gb:24.95
|
| 142 |
+
Training: 35%|ββββ | 20800/60000 [8:28:24<17:48:36, 1.64s/step]INFO 2026-07-22 10:36:33 ot_train.py:649 step:21K smpl:1M ep:30K epch:6.10 loss:0.060 grdn:0.339 lr:7.8e-05 updt_s:0.802 data_s:0.756 smp/s:41 mem_gb:24.95
|
| 143 |
+
Training: 35%|ββββ | 21000/60000 [8:33:41<16:52:48, 1.56s/step]INFO 2026-07-22 10:41:50 ot_train.py:649 step:21K smpl:1M ep:30K epch:6.15 loss:0.058 grdn:0.330 lr:7.8e-05 updt_s:0.816 data_s:0.764 smp/s:40 mem_gb:24.95
|
| 144 |
+
Training: 35%|ββββ | 21200/60000 [8:38:56<16:39:14, 1.55s/step]INFO 2026-07-22 10:47:06 ot_train.py:649 step:21K smpl:1M ep:31K epch:6.21 loss:0.057 grdn:0.327 lr:7.7e-05 updt_s:0.816 data_s:0.758 smp/s:41 mem_gb:24.95
|
| 145 |
+
Training: 36%|ββββ | 21400/60000 [8:44:13<18:56:56, 1.77s/step]INFO 2026-07-22 10:52:23 ot_train.py:649 step:21K smpl:1M ep:31K epch:6.27 loss:0.057 grdn:0.318 lr:7.7e-05 updt_s:0.820 data_s:0.761 smp/s:40 mem_gb:24.95
|
| 146 |
+
Training: 36%|ββββ | 21600/60000 [8:49:33<16:29:38, 1.55s/step]INFO 2026-07-22 10:57:42 ot_train.py:649 step:22K smpl:1M ep:31K epch:6.33 loss:0.058 grdn:0.328 lr:7.6e-05 updt_s:0.821 data_s:0.771 smp/s:40 mem_gb:24.95
|
| 147 |
+
Training: 36%|ββββ | 21800/60000 [8:54:53<17:50:20, 1.68s/step]INFO 2026-07-22 11:03:03 ot_train.py:649 step:22K smpl:1M ep:31K epch:6.39 loss:0.057 grdn:0.320 lr:7.6e-05 updt_s:0.829 data_s:0.771 smp/s:40 mem_gb:24.95
|
| 148 |
+
Training: 37%|ββββ | 22000/60000 [9:00:10<15:56:38, 1.51s/step]INFO 2026-07-22 11:08:19 ot_train.py:649 step:22K smpl:1M ep:32K epch:6.45 loss:0.058 grdn:0.331 lr:7.5e-05 updt_s:0.816 data_s:0.762 smp/s:41 mem_gb:24.95
|
| 149 |
+
Training: 37%|ββββ | 22200/60000 [9:05:08<15:59:10, 1.52s/step]INFO 2026-07-22 11:13:18 ot_train.py:649 step:22K smpl:1M ep:32K epch:6.51 loss:0.058 grdn:0.335 lr:7.5e-05 updt_s:0.797 data_s:0.691 smp/s:43 mem_gb:24.95
|
| 150 |
+
Training: 37%|ββββ | 22400/60000 [9:09:46<14:22:35, 1.38s/step]INFO 2026-07-22 11:17:56 ot_train.py:649 step:22K smpl:1M ep:32K epch:6.57 loss:0.058 grdn:0.335 lr:7.4e-05 updt_s:0.744 data_s:0.642 smp/s:46 mem_gb:24.95
|
| 151 |
+
Training: 38%|ββββ | 22600/60000 [9:14:26<15:09:36, 1.46s/step]INFO 2026-07-22 11:22:35 ot_train.py:649 step:23K smpl:1M ep:33K epch:6.62 loss:0.058 grdn:0.338 lr:7.4e-05 updt_s:0.746 data_s:0.649 smp/s:46 mem_gb:24.93
|
| 152 |
+
Training: 38%|ββββ | 22800/60000 [9:19:16<14:36:22, 1.41s/step]INFO 2026-07-22 11:27:25 ot_train.py:649 step:23K smpl:1M ep:33K epch:6.68 loss:0.054 grdn:0.317 lr:7.3e-05 updt_s:0.781 data_s:0.666 smp/s:44 mem_gb:24.95
|
| 153 |
+
Training: 38%|ββββ | 23000/60000 [9:24:06<14:43:05, 1.43s/step]INFO 2026-07-22 11:32:16 ot_train.py:649 step:23K smpl:1M ep:33K epch:6.74 loss:0.054 grdn:0.337 lr:7.3e-05 updt_s:0.771 data_s:0.676 smp/s:44 mem_gb:24.95
|
| 154 |
+
Training: 39%|ββββ | 23200/60000 [9:28:41<14:08:55, 1.38s/step]INFO 2026-07-22 11:36:51 ot_train.py:649 step:23K smpl:1M ep:34K epch:6.80 loss:0.054 grdn:0.333 lr:7.2e-05 updt_s:0.731 data_s:0.643 smp/s:47 mem_gb:24.95
|
| 155 |
+
Training: 39%|ββββ | 23400/60000 [9:33:24<14:19:55, 1.41s/step]INFO 2026-07-22 11:41:34 ot_train.py:649 step:23K smpl:1M ep:34K epch:6.86 loss:0.054 grdn:0.347 lr:7.2e-05 updt_s:0.748 data_s:0.662 smp/s:45 mem_gb:24.95
|
| 156 |
+
Training: 39%|ββββ | 23600/60000 [9:38:08<14:07:41, 1.40s/step]INFO 2026-07-22 11:46:18 ot_train.py:649 step:24K smpl:2M ep:34K epch:6.92 loss:0.055 grdn:0.341 lr:7.1e-05 updt_s:0.758 data_s:0.660 smp/s:45 mem_gb:24.95
|
| 157 |
+
Training: 40%|ββββ | 23800/60000 [9:42:50<14:15:54, 1.42s/step]INFO 2026-07-22 11:51:00 ot_train.py:649 step:24K smpl:2M ep:34K epch:6.98 loss:0.053 grdn:0.325 lr:7.1e-05 updt_s:0.749 data_s:0.658 smp/s:45 mem_gb:24.95
|
| 158 |
+
Training: 40%|ββββ | 24000/60000 [9:47:47<14:51:34, 1.49s/step]INFO 2026-07-22 11:55:56 ot_train.py:649 step:24K smpl:2M ep:35K epch:7.03 loss:0.053 grdn:0.337 lr:7.0e-05 updt_s:0.783 data_s:0.696 smp/s:43 mem_gb:24.95
|
| 159 |
+
Training: 40%|ββββ | 24200/60000 [9:52:49<13:53:05, 1.40s/step]INFO 2026-07-22 12:00:58 ot_train.py:649 step:24K smpl:2M ep:35K epch:7.09 loss:0.053 grdn:0.337 lr:7.0e-05 updt_s:0.785 data_s:0.721 smp/s:42 mem_gb:24.95
|
| 160 |
+
Training: 41%|ββββ | 24400/60000 [9:57:35<13:13:12, 1.34s/step]INFO 2026-07-22 12:05:45 ot_train.py:649 step:24K smpl:2M ep:35K epch:7.15 loss:0.052 grdn:0.329 lr:6.9e-05 updt_s:0.769 data_s:0.659 smp/s:45 mem_gb:24.95
|
| 161 |
+
Training: 41%|ββββ | 24600/60000 [10:02:14<14:07:27, 1.44s/step]INFO 2026-07-22 12:10:24 ot_train.py:649 step:25K smpl:2M ep:36K epch:7.21 loss:0.053 grdn:0.329 lr:6.9e-05 updt_s:0.733 data_s:0.659 smp/s:46 mem_gb:24.95
|
| 162 |
+
Training: 41%|βββββ | 24800/60000 [10:06:54<14:00:23, 1.43s/step]INFO 2026-07-22 12:15:04 ot_train.py:649 step:25K smpl:2M ep:36K epch:7.27 loss:0.052 grdn:0.339 lr:6.8e-05 updt_s:0.735 data_s:0.662 smp/s:46 mem_gb:24.95
|
| 163 |
+
Training: 42%|βββββ | 25000/60000 [10:11:38<13:47:27, 1.42s/step]INFO 2026-07-22 12:19:47 ot_train.py:649 step:25K smpl:2M ep:36K epch:7.33 loss:0.051 grdn:0.332 lr:6.8e-05 updt_s:0.747 data_s:0.668 smp/s:45 mem_gb:24.93
|
| 164 |
+
Training: 42%|βββββ | 25200/60000 [10:16:28<13:49:56, 1.43s/step]INFO 2026-07-22 12:24:38 ot_train.py:649 step:25K smpl:2M ep:36K epch:7.39 loss:0.050 grdn:0.336 lr:6.7e-05 updt_s:0.764 data_s:0.684 smp/s:44 mem_gb:24.95
|
| 165 |
+
Training: 42%|βββββ | 25400/60000 [10:21:16<13:52:20, 1.44s/step]INFO 2026-07-22 12:29:26 ot_train.py:649 step:25K smpl:2M ep:37K epch:7.44 loss:0.048 grdn:0.321 lr:6.7e-05 updt_s:0.754 data_s:0.685 smp/s:44 mem_gb:24.95
|
| 166 |
+
Training: 43%|βββββ | 25600/60000 [10:26:01<15:14:26, 1.59s/step]INFO 2026-07-22 12:34:10 ot_train.py:649 step:26K smpl:2M ep:37K epch:7.50 loss:0.051 grdn:0.339 lr:6.6e-05 updt_s:0.754 data_s:0.664 smp/s:45 mem_gb:24.95
|
| 167 |
+
Training: 43%|βββββ | 25800/60000 [10:30:40<12:58:35, 1.37s/step]INFO 2026-07-22 12:38:50 ot_train.py:649 step:26K smpl:2M ep:37K epch:7.56 loss:0.049 grdn:0.333 lr:6.6e-05 updt_s:0.730 data_s:0.664 smp/s:46 mem_gb:24.95
|
| 168 |
+
Training: 43%|βββββ | 26000/60000 [10:35:21<13:05:42, 1.39s/step]INFO 2026-07-22 12:43:31 ot_train.py:649 step:26K smpl:2M ep:38K epch:7.62 loss:0.049 grdn:0.337 lr:6.5e-05 updt_s:0.733 data_s:0.668 smp/s:46 mem_gb:24.95
|
| 169 |
+
Training: 44%|βββββ | 26200/60000 [10:40:03<13:10:27, 1.40s/step]INFO 2026-07-22 12:48:12 ot_train.py:649 step:26K smpl:2M ep:38K epch:7.68 loss:0.049 grdn:0.316 lr:6.5e-05 updt_s:0.747 data_s:0.657 smp/s:46 mem_gb:24.95
|
| 170 |
+
Training: 44%|βββββ | 26400/60000 [10:44:46<13:42:28, 1.47s/step]INFO 2026-07-22 12:52:56 ot_train.py:649 step:26K smpl:2M ep:38K epch:7.74 loss:0.047 grdn:0.313 lr:6.4e-05 updt_s:0.746 data_s:0.668 smp/s:45 mem_gb:24.95
|
| 171 |
+
Training: 44%|βββββ | 26600/60000 [10:49:11<11:23:30, 1.23s/step]INFO 2026-07-22 12:57:20 ot_train.py:649 step:27K smpl:2M ep:38K epch:7.80 loss:0.048 grdn:0.316 lr:6.4e-05 updt_s:0.705 data_s:0.615 smp/s:48 mem_gb:24.95
|
| 172 |
+
Training: 45%|βββββ | 26800/60000 [10:53:16<11:14:11, 1.22s/step]INFO 2026-07-22 13:01:26 ot_train.py:649 step:27K smpl:2M ep:39K epch:7.85 loss:0.047 grdn:0.322 lr:6.3e-05 updt_s:0.662 data_s:0.562 smp/s:52 mem_gb:24.95
|
| 173 |
+
Training: 45%|βββββ | 27000/60000 [10:57:29<11:07:30, 1.21s/step]INFO 2026-07-22 13:05:39 ot_train.py:649 step:27K smpl:2M ep:39K epch:7.91 loss:0.048 grdn:0.317 lr:6.3e-05 updt_s:0.686 data_s:0.577 smp/s:51 mem_gb:24.95
|
| 174 |
+
Training: 45%|βββββ | 27200/60000 [11:01:39<11:23:29, 1.25s/step]INFO 2026-07-22 13:09:48 ot_train.py:649 step:27K smpl:2M ep:39K epch:7.97 loss:0.046 grdn:0.317 lr:6.2e-05 updt_s:0.664 data_s:0.581 smp/s:51 mem_gb:24.93
|
| 175 |
+
Training: 46%|βββββ | 27400/60000 [11:06:09<12:24:24, 1.37s/step]INFO 2026-07-22 13:14:18 ot_train.py:649 step:27K smpl:2M ep:40K epch:8.03 loss:0.045 grdn:0.324 lr:6.1e-05 updt_s:0.716 data_s:0.631 smp/s:48 mem_gb:24.95
|
| 176 |
+
Training: 46%|βββββ | 27600/60000 [11:10:43<11:57:25, 1.33s/step]INFO 2026-07-22 13:18:53 ot_train.py:649 step:28K smpl:2M ep:40K epch:8.09 loss:0.045 grdn:0.317 lr:6.1e-05 updt_s:0.731 data_s:0.641 smp/s:47 mem_gb:24.95
|
| 177 |
+
Training: 46%|βββββ | 27800/60000 [11:15:17<12:01:31, 1.34s/step]INFO 2026-07-22 13:23:26 ot_train.py:649 step:28K smpl:2M ep:40K epch:8.15 loss:0.043 grdn:0.322 lr:6.0e-05 updt_s:0.717 data_s:0.645 smp/s:47 mem_gb:24.95
|
| 178 |
+
Training: 47%|βββββ | 28000/60000 [11:19:43<11:21:11, 1.28s/step]INFO 2026-07-22 13:27:53 ot_train.py:649 step:28K smpl:2M ep:40K epch:8.21 loss:0.046 grdn:0.343 lr:6.0e-05 updt_s:0.697 data_s:0.633 smp/s:48 mem_gb:24.95
|
| 179 |
+
Training: 47%|βββββ | 28200/60000 [11:24:16<12:02:57, 1.36s/step]INFO 2026-07-22 13:32:26 ot_train.py:649 step:28K smpl:2M ep:41K epch:8.26 loss:0.044 grdn:0.333 lr:5.9e-05 updt_s:0.716 data_s:0.648 smp/s:47 mem_gb:24.95
|
| 180 |
+
Training: 47%|βββββ | 28400/60000 [11:29:00<12:03:38, 1.37s/step]INFO 2026-07-22 13:37:10 ot_train.py:649 step:28K smpl:2M ep:41K epch:8.32 loss:0.044 grdn:0.312 lr:5.9e-05 updt_s:0.755 data_s:0.659 smp/s:45 mem_gb:24.95
|
| 181 |
+
Training: 48%|βββββ | 28600/60000 [11:33:42<12:17:06, 1.41s/step]INFO 2026-07-22 13:41:52 ot_train.py:649 step:29K smpl:2M ep:41K epch:8.38 loss:0.045 grdn:0.337 lr:5.8e-05 updt_s:0.728 data_s:0.679 smp/s:45 mem_gb:24.95
|
| 182 |
+
Training: 48%|βββββ | 28800/60000 [11:38:37<12:49:31, 1.48s/step]INFO 2026-07-22 13:46:47 ot_train.py:649 step:29K smpl:2M ep:42K epch:8.44 loss:0.044 grdn:0.331 lr:5.8e-05 updt_s:0.760 data_s:0.713 smp/s:43 mem_gb:24.95
|
| 183 |
+
Training: 48%|βββββ | 29000/60000 [11:43:25<12:07:12, 1.41s/step]INFO 2026-07-22 13:51:35 ot_train.py:649 step:29K smpl:2M ep:42K epch:8.50 loss:0.044 grdn:0.317 lr:5.7e-05 updt_s:0.737 data_s:0.699 smp/s:45 mem_gb:24.95
|
| 184 |
+
Training: 49%|βββββ | 29200/60000 [11:48:14<12:07:07, 1.42s/step]INFO 2026-07-22 13:56:23 ot_train.py:649 step:29K smpl:2M ep:42K epch:8.56 loss:0.044 grdn:0.308 lr:5.7e-05 updt_s:0.756 data_s:0.683 smp/s:44 mem_gb:24.95
|
| 185 |
+
Training: 49%|βββββ | 29400/60000 [11:53:00<11:40:53, 1.37s/step]INFO 2026-07-22 14:01:09 ot_train.py:649 step:29K smpl:2M ep:42K epch:8.62 loss:0.043 grdn:0.339 lr:5.6e-05 updt_s:0.743 data_s:0.685 smp/s:45 mem_gb:24.93
|
| 186 |
+
Training: 49%|βββββ | 29600/60000 [11:57:47<12:15:25, 1.45s/step]INFO 2026-07-22 14:05:56 ot_train.py:649 step:30K smpl:2M ep:43K epch:8.68 loss:0.042 grdn:0.310 lr:5.5e-05 updt_s:0.754 data_s:0.677 smp/s:45 mem_gb:24.95
|
| 187 |
+
Training: 50%|βββββ | 29800/60000 [12:03:05<11:55:16, 1.42s/step]INFO 2026-07-22 14:11:14 ot_train.py:649 step:30K smpl:2M ep:43K epch:8.73 loss:0.040 grdn:0.304 lr:5.5e-05 updt_s:0.765 data_s:0.820 smp/s:40 mem_gb:24.95
|
| 188 |
+
Training: 50%|βββββ | 30000/60000 [12:08:10<16:29:57, 1.98s/step]INFO 2026-07-22 14:16:20 ot_train.py:649 step:30K smpl:2M ep:43K epch:8.79 loss:0.040 grdn:0.327 lr:5.4e-05 updt_s:0.751 data_s:0.772 smp/s:42 mem_gb:24.95
|
| 189 |
+
INFO 2026-07-22 14:16:20 ot_train.py:694 Checkpoint policy after step 30000
|
| 190 |
+
INFO 2026-07-22 14:16:47 in_yaml.py:1270 Saved config, action-space, prompt, and phase provenance in /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K and /home/ext_minje/groot_insight/Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/checkpoints/030000
|
| 191 |
+
Training: 50%|βββββ | 30200/60000 [12:14:20<14:38:30, 1.77s/step]INFO 2026-07-22 14:22:30 ot_train.py:649 step:30K smpl:2M ep:44K epch:8.85 loss:0.041 grdn:0.318 lr:5.4e-05 updt_s:0.723 data_s:0.986 smp/s:37 mem_gb:24.95
|
| 192 |
+
Training: 51%|βββββ | 30400/60000 [12:23:41<13:42:34, 1.67s/step]INFO 2026-07-22 14:31:51 ot_train.py:649 step:30K smpl:2M ep:44K epch:8.91 loss:0.039 grdn:0.292 lr:5.3e-05 updt_s:0.794 data_s:2.009 smp/s:23 mem_gb:24.95
|
| 193 |
+
Training: 51%|βββββ | 30600/60000 [12:28:54<12:32:43, 1.54s/step]INFO 2026-07-22 14:37:04 ot_train.py:649 step:31K smpl:2M ep:44K epch:8.97 loss:0.041 grdn:0.328 lr:5.3e-05 updt_s:0.792 data_s:0.771 smp/s:41 mem_gb:24.95
|
| 194 |
+
Training: 51%|ββββββ | 30800/60000 [12:34:40<11:48:47, 1.46s/step]INFO 2026-07-22 14:42:49 ot_train.py:649 step:31K smpl:2M ep:45K epch:9.03 loss:0.040 grdn:0.321 lr:5.2e-05 updt_s:0.799 data_s:0.923 smp/s:37 mem_gb:24.95
|
| 195 |
+
Training: 52%|ββββββ | 31000/60000 [12:40:41<15:23:00, 1.91s/step]INFO 2026-07-22 14:48:51 ot_train.py:649 step:31K smpl:2M ep:45K epch:9.09 loss:0.040 grdn:0.316 lr:5.2e-05 updt_s:0.765 data_s:1.039 smp/s:35 mem_gb:24.95
|
| 196 |
+
Training: 52%|ββββββ | 31200/60000 [12:46:14<12:47:05, 1.60s/step]INFO 2026-07-22 14:54:24 ot_train.py:649 step:31K smpl:2M ep:45K epch:9.14 loss:0.038 grdn:0.312 lr:5.1e-05 updt_s:0.765 data_s:0.897 smp/s:39 mem_gb:24.95
|
| 197 |
+
Training: 52%|ββββββ | 31400/60000 [12:51:45<12:19:33, 1.55s/step]INFO 2026-07-22 14:59:55 ot_train.py:649 step:31K smpl:2M ep:45K epch:9.20 loss:0.040 grdn:0.315 lr:5.1e-05 updt_s:0.775 data_s:0.877 smp/s:39 mem_gb:24.95
|
| 198 |
+
Training: 53%|ββββββ | 31600/60000 [12:57:10<15:53:42, 2.01s/step]INFO 2026-07-22 15:05:20 ot_train.py:649 step:32K smpl:2M ep:46K epch:9.26 loss:0.039 grdn:0.293 lr:5.0e-05 updt_s:0.767 data_s:0.854 smp/s:39 mem_gb:24.93
|
| 199 |
+
Training: 53%|ββββββ | 31800/60000 [13:02:41<11:41:35, 1.49s/step]INFO 2026-07-22 15:10:50 ot_train.py:649 step:32K smpl:2M ep:46K epch:9.32 loss:0.037 grdn:0.300 lr:4.9e-05 updt_s:0.754 data_s:0.896 smp/s:39 mem_gb:24.95
|
| 200 |
+
Training: 53%|ββββββ | 32000/60000 [13:08:15<13:24:14, 1.72s/step]INFO 2026-07-22 15:16:24 ot_train.py:649 step:32K smpl:2M ep:46K epch:9.38 loss:0.037 grdn:0.316 lr:4.9e-05 updt_s:0.735 data_s:0.930 smp/s:38 mem_gb:24.95
|
| 201 |
+
Training: 54%|ββββββ | 32200/60000 [13:13:51<12:35:49, 1.63s/step]INFO 2026-07-22 15:22:01 ot_train.py:649 step:32K smpl:2M ep:47K epch:9.44 loss:0.038 grdn:0.312 lr:4.8e-05 updt_s:0.733 data_s:0.948 smp/s:38 mem_gb:24.95
|
| 202 |
+
Training: 54%|ββββββ | 32400/60000 [13:19:34<12:09:59, 1.59s/step]INFO 2026-07-22 15:27:43 ot_train.py:649 step:32K smpl:2M ep:47K epch:9.50 loss:0.037 grdn:0.309 lr:4.8e-05 updt_s:0.724 data_s:0.984 smp/s:37 mem_gb:24.95
|
| 203 |
+
Training: 54%|ββββββ | 32600/60000 [13:24:45<11:38:53, 1.53s/step]INFO 2026-07-22 15:32:54 ot_train.py:649 step:33K smpl:2M ep:47K epch:9.55 loss:0.037 grdn:0.300 lr:4.7e-05 updt_s:0.731 data_s:0.821 smp/s:41 mem_gb:24.95
|
| 204 |
+
Training: 55%|ββββββ | 32800/60000 [13:30:01<11:39:49, 1.54s/step]INFO 2026-07-22 15:38:11 ot_train.py:649 step:33K smpl:2M ep:47K epch:9.61 loss:0.037 grdn:0.301 lr:4.7e-05 updt_s:0.765 data_s:0.814 smp/s:41 mem_gb:24.95
|
| 205 |
+
Training: 55%|ββββββ | 33000/60000 [13:35:27<23:09:11, 3.09s/step]INFO 2026-07-22 15:43:36 ot_train.py:649 step:33K smpl:2M ep:48K epch:9.67 loss:0.036 grdn:0.293 lr:4.6e-05 updt_s:0.750 data_s:0.875 smp/s:39 mem_gb:24.95
|
| 206 |
+
Training: 55%|ββββββ | 33200/60000 [13:41:11<11:03:16, 1.48s/step]INFO 2026-07-22 15:49:21 ot_train.py:649 step:33K smpl:2M ep:48K epch:9.73 loss:0.036 grdn:0.291 lr:4.6e-05 updt_s:0.729 data_s:0.988 smp/s:37 mem_gb:24.95
|
| 207 |
+
Training: 56%|ββββββ | 33400/60000 [13:46:14<11:06:33, 1.50s/step]INFO 2026-07-22 15:54:23 ot_train.py:649 step:33K smpl:2M ep:48K epch:9.79 loss:0.036 grdn:0.301 lr:4.5e-05 updt_s:0.743 data_s:0.765 smp/s:42 mem_gb:24.95
|
| 208 |
+
Training: 56%|ββββββ | 33600/60000 [13:51:33<10:32:19, 1.44s/step]INFO 2026-07-22 15:59:42 ot_train.py:649 step:34K smpl:2M ep:49K epch:9.85 loss:0.036 grdn:0.308 lr:4.4e-05 updt_s:0.732 data_s:0.862 smp/s:40 mem_gb:24.95
|
| 209 |
+
Training: 56%|ββββββ | 33800/60000 [13:56:35<11:34:13, 1.59s/step]INFO 2026-07-22 16:04:44 ot_train.py:649 step:34K smpl:2M ep:49K epch:9.91 loss:0.036 grdn:0.296 lr:4.4e-05 updt_s:0.754 data_s:0.752 smp/s:42 mem_gb:24.95
|
| 210 |
+
Training: 57%|ββββββ | 34000/60000 [14:01:45<10:52:43, 1.51s/step]INFO 2026-07-22 16:09:55 ot_train.py:649 step:34K smpl:2M ep:49K epch:9.96 loss:0.035 grdn:0.308 lr:4.3e-05 updt_s:0.767 data_s:0.782 smp/s:41 mem_gb:24.93
|
| 211 |
+
Training: 57%|ββββββ | 34200/60000 [14:06:52<10:52:58, 1.52s/step]INFO 2026-07-22 16:15:02 ot_train.py:649 step:34K smpl:2M ep:49K epch:10.02 loss:0.033 grdn:0.295 lr:4.3e-05 updt_s:0.762 data_s:0.769 smp/s:42 mem_gb:24.95
|
| 212 |
+
Training: 57%|ββββββ | 34400/60000 [14:12:00<9:51:10, 1.39s/step] INFO 2026-07-22 16:20:09 ot_train.py:649 step:34K smpl:2M ep:50K epch:10.08 loss:0.032 grdn:0.290 lr:4.2e-05 updt_s:0.750 data_s:0.784 smp/s:42 mem_gb:24.95
|
| 213 |
+
Training: 58%|ββββββ | 34600/60000 [14:16:08<8:43:24, 1.24s/step]INFO 2026-07-22 16:24:18 ot_train.py:649 step:35K smpl:2M ep:50K epch:10.14 loss:0.033 grdn:0.310 lr:4.2e-05 updt_s:0.646 data_s:0.593 smp/s:52 mem_gb:24.95
|
| 214 |
+
Training: 58%|ββββββ | 34800/60000 [14:20:33<8:46:56, 1.25s/step]INFO 2026-07-22 16:28:42 ot_train.py:649 step:35K smpl:2M ep:50K epch:10.20 loss:0.034 grdn:0.303 lr:4.1e-05 updt_s:0.658 data_s:0.661 smp/s:49 mem_gb:24.95
|
| 215 |
+
Training: 58%|ββββββ | 35000/60000 [14:25:40<8:45:42, 1.26s/step]INFO 2026-07-22 16:33:50 ot_train.py:649 step:35K smpl:2M ep:51K epch:10.26 loss:0.033 grdn:0.291 lr:4.1e-05 updt_s:0.672 data_s:0.865 smp/s:42 mem_gb:24.95
|
| 216 |
+
Training: 59%|ββββββ | 35200/60000 [14:30:01<8:31:39, 1.24s/step]INFO 2026-07-22 16:38:11 ot_train.py:649 step:35K smpl:2M ep:51K epch:10.32 loss:0.033 grdn:0.298 lr:4.0e-05 updt_s:0.668 data_s:0.633 smp/s:49 mem_gb:24.95
|
| 217 |
+
Training: 59%|ββββββ | 35400/60000 [14:34:08<8:14:30, 1.21s/step]INFO 2026-07-22 16:42:17 ot_train.py:649 step:35K smpl:2M ep:51K epch:10.38 loss:0.032 grdn:0.290 lr:4.0e-05 updt_s:0.649 data_s:0.581 smp/s:52 mem_gb:24.95
|
| 218 |
+
Training: 59%|ββββββ | 35600/60000 [14:38:14<8:27:21, 1.25s/step]INFO 2026-07-22 16:46:24 ot_train.py:649 step:36K smpl:2M ep:51K epch:10.43 loss:0.033 grdn:0.296 lr:3.9e-05 updt_s:0.650 data_s:0.579 smp/s:52 mem_gb:24.95
|
| 219 |
+
Training: 60%|ββββββ | 35800/60000 [14:42:21<8:19:30, 1.24s/step]INFO 2026-07-22 16:50:30 ot_train.py:649 step:36K smpl:2M ep:52K epch:10.49 loss:0.032 grdn:0.287 lr:3.9e-05 updt_s:0.651 data_s:0.579 smp/s:52 mem_gb:24.95
|
| 220 |
+
Training: 60%|ββββββ | 36000/60000 [14:46:25<8:23:06, 1.26s/step]INFO 2026-07-22 16:54:35 ot_train.py:649 step:36K smpl:2M ep:52K epch:10.55 loss:0.031 grdn:0.274 lr:3.8e-05 updt_s:0.652 data_s:0.567 smp/s:52 mem_gb:24.95
|
| 221 |
+
Training: 60%|ββββββ | 36200/60000 [14:50:31<8:09:53, 1.24s/step]INFO 2026-07-22 16:58:40 ot_train.py:649 step:36K smpl:2M ep:52K epch:10.61 loss:0.031 grdn:0.286 lr:3.7e-05 updt_s:0.654 data_s:0.572 smp/s:52 mem_gb:24.93
|
| 222 |
+
Training: 61%|ββββββ | 36400/60000 [14:54:38<8:13:43, 1.26s/step]INFO 2026-07-22 17:02:48 ot_train.py:649 step:36K smpl:2M ep:53K epch:10.67 loss:0.031 grdn:0.297 lr:3.7e-05 updt_s:0.665 data_s:0.570 smp/s:52 mem_gb:24.95
|
| 223 |
+
Training: 61%|ββββββ | 36600/60000 [14:58:43<7:48:04, 1.20s/step]INFO 2026-07-22 17:06:52 ot_train.py:649 step:37K smpl:2M ep:53K epch:10.73 loss:0.031 grdn:0.287 lr:3.6e-05 updt_s:0.649 data_s:0.571 smp/s:52 mem_gb:24.95
|
| 224 |
+
Training: 61%|βββββββ | 36800/60000 [15:02:46<7:52:01, 1.22s/step]INFO 2026-07-22 17:10:55 ot_train.py:649 step:37K smpl:2M ep:53K epch:10.79 loss:0.029 grdn:0.269 lr:3.6e-05 updt_s:0.651 data_s:0.562 smp/s:53 mem_gb:24.95
|
| 225 |
+
Training: 62%|βββββββ | 37000/60000 [15:06:47<7:31:09, 1.18s/step]INFO 2026-07-22 17:14:56 ot_train.py:649 step:37K smpl:2M ep:53K epch:10.84 loss:0.030 grdn:0.280 lr:3.5e-05 updt_s:0.644 data_s:0.560 smp/s:53 mem_gb:24.95
|
| 226 |
+
Training: 62%|βββββββ | 37023/60000 [15:07:15<7:52:26, 1.23s/step]
|
Abs_6D/DiT_Layer/8/RSCLstyle_cosine_60K/wandb/run-20260722_020800-rd8tz18f/logs/debug-internal.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|