Delete checkpoint-9000
Browse files- checkpoint-9000/config.json +0 -64
- checkpoint-9000/experiment_cfg/metadata.json +0 -195
- checkpoint-9000/model-00001-of-00002.safetensors +0 -3
- checkpoint-9000/model-00002-of-00002.safetensors +0 -3
- checkpoint-9000/model.safetensors.index.json +0 -0
- checkpoint-9000/optimizer.pt +0 -3
- checkpoint-9000/rng_state.pth +0 -3
- checkpoint-9000/scheduler.pt +0 -3
- checkpoint-9000/trainer_state.json +0 -0
checkpoint-9000/config.json
DELETED
|
@@ -1,64 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"action_dim": 32,
|
| 3 |
-
"action_head_cfg": {
|
| 4 |
-
"action_dim": 32,
|
| 5 |
-
"action_horizon": 16,
|
| 6 |
-
"add_pos_embed": true,
|
| 7 |
-
"backbone_embedding_dim": 2048,
|
| 8 |
-
"diffusion_model_cfg": {
|
| 9 |
-
"attention_head_dim": 48,
|
| 10 |
-
"cross_attention_dim": 2048,
|
| 11 |
-
"dropout": 0.2,
|
| 12 |
-
"final_dropout": true,
|
| 13 |
-
"interleave_self_attention": true,
|
| 14 |
-
"norm_type": "ada_norm",
|
| 15 |
-
"num_attention_heads": 32,
|
| 16 |
-
"num_layers": 16,
|
| 17 |
-
"output_dim": 1024,
|
| 18 |
-
"positional_embeddings": null
|
| 19 |
-
},
|
| 20 |
-
"hidden_size": 1024,
|
| 21 |
-
"input_embedding_dim": 1536,
|
| 22 |
-
"max_action_dim": 32,
|
| 23 |
-
"max_state_dim": 64,
|
| 24 |
-
"model_dtype": "float32",
|
| 25 |
-
"noise_beta_alpha": 1.5,
|
| 26 |
-
"noise_beta_beta": 1.0,
|
| 27 |
-
"noise_s": 0.999,
|
| 28 |
-
"num_inference_timesteps": 4,
|
| 29 |
-
"num_target_vision_tokens": 32,
|
| 30 |
-
"num_timestep_buckets": 1000,
|
| 31 |
-
"tune_diffusion_model": true,
|
| 32 |
-
"tune_projector": true,
|
| 33 |
-
"use_vlln": true,
|
| 34 |
-
"vl_self_attention_cfg": {
|
| 35 |
-
"attention_head_dim": 64,
|
| 36 |
-
"dropout": 0.2,
|
| 37 |
-
"final_dropout": true,
|
| 38 |
-
"num_attention_heads": 32,
|
| 39 |
-
"num_layers": 4,
|
| 40 |
-
"positional_embeddings": null
|
| 41 |
-
}
|
| 42 |
-
},
|
| 43 |
-
"action_horizon": 16,
|
| 44 |
-
"architectures": [
|
| 45 |
-
"GR00T_N1_5"
|
| 46 |
-
],
|
| 47 |
-
"attn_implementation": null,
|
| 48 |
-
"backbone_cfg": {
|
| 49 |
-
"eagle_path": "NVEagle/eagle_er-qwen3_1_7B-Siglip2_400M_stage1_5_128gpu_er_v7_1mlp_nops",
|
| 50 |
-
"load_bf16": false,
|
| 51 |
-
"project_to_dim": null,
|
| 52 |
-
"reproject_vision": false,
|
| 53 |
-
"select_layer": 12,
|
| 54 |
-
"tune_llm": false,
|
| 55 |
-
"tune_visual": true,
|
| 56 |
-
"use_flash_attention": true
|
| 57 |
-
},
|
| 58 |
-
"compute_dtype": "bfloat16",
|
| 59 |
-
"hidden_size": 2048,
|
| 60 |
-
"model_dtype": "float32",
|
| 61 |
-
"model_type": "gr00t_n1_5",
|
| 62 |
-
"torch_dtype": "bfloat16",
|
| 63 |
-
"transformers_version": "4.51.3"
|
| 64 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/experiment_cfg/metadata.json
DELETED
|
@@ -1,195 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"new_embodiment": {
|
| 3 |
-
"statistics": {
|
| 4 |
-
"state": {
|
| 5 |
-
"single_arm": {
|
| 6 |
-
"max": [
|
| 7 |
-
31.584779739379883,
|
| 8 |
-
46.67191696166992,
|
| 9 |
-
99.81981658935547,
|
| 10 |
-
95.37869262695312,
|
| 11 |
-
-44.607330322265625
|
| 12 |
-
],
|
| 13 |
-
"min": [
|
| 14 |
-
-35.05725860595703,
|
| 15 |
-
-98.50334930419922,
|
| 16 |
-
-25.405405044555664,
|
| 17 |
-
-11.339323997497559,
|
| 18 |
-
-62.04188537597656
|
| 19 |
-
],
|
| 20 |
-
"mean": [
|
| 21 |
-
5.172351822039814,
|
| 22 |
-
-31.90936843108956,
|
| 23 |
-
44.84144609128404,
|
| 24 |
-
25.06023244569546,
|
| 25 |
-
-51.23305436968347
|
| 26 |
-
],
|
| 27 |
-
"std": [
|
| 28 |
-
15.328154290181528,
|
| 29 |
-
44.35800487695137,
|
| 30 |
-
38.24853805975966,
|
| 31 |
-
14.809518746264933,
|
| 32 |
-
2.7295327930239175
|
| 33 |
-
],
|
| 34 |
-
"q01": [
|
| 35 |
-
-10.875112238811683,
|
| 36 |
-
-97.79692490173177,
|
| 37 |
-
-4.233867185986334,
|
| 38 |
-
12.802839856980988,
|
| 39 |
-
-53.25761131186619
|
| 40 |
-
],
|
| 41 |
-
"q99": [
|
| 42 |
-
27.84463228153257,
|
| 43 |
-
21.49041878209509,
|
| 44 |
-
99.28034884491058,
|
| 45 |
-
38.657944851567294,
|
| 46 |
-
-49.089739835646256
|
| 47 |
-
]
|
| 48 |
-
},
|
| 49 |
-
"gripper": {
|
| 50 |
-
"max": [
|
| 51 |
-
68.24457550048828
|
| 52 |
-
],
|
| 53 |
-
"min": [
|
| 54 |
-
7.297830581665039
|
| 55 |
-
],
|
| 56 |
-
"mean": [
|
| 57 |
-
15.647195632321983
|
| 58 |
-
],
|
| 59 |
-
"std": [
|
| 60 |
-
12.412345851759543
|
| 61 |
-
],
|
| 62 |
-
"q01": [
|
| 63 |
-
8.452010638589131
|
| 64 |
-
],
|
| 65 |
-
"q99": [
|
| 66 |
-
49.39492147723287
|
| 67 |
-
]
|
| 68 |
-
}
|
| 69 |
-
},
|
| 70 |
-
"action": {
|
| 71 |
-
"single_arm": {
|
| 72 |
-
"max": [
|
| 73 |
-
31.858407974243164,
|
| 74 |
-
46.33333206176758,
|
| 75 |
-
100.0,
|
| 76 |
-
95.79627990722656,
|
| 77 |
-
-44.4587287902832
|
| 78 |
-
],
|
| 79 |
-
"min": [
|
| 80 |
-
-35.1769905090332,
|
| 81 |
-
-99.66666412353516,
|
| 82 |
-
-29.768917083740234,
|
| 83 |
-
-12.611156463623047,
|
| 84 |
-
-61.89251708984375
|
| 85 |
-
],
|
| 86 |
-
"mean": [
|
| 87 |
-
5.338037511643458,
|
| 88 |
-
-33.247942721346924,
|
| 89 |
-
41.79598039744411,
|
| 90 |
-
24.384786050823003,
|
| 91 |
-
-51.2632397695491
|
| 92 |
-
],
|
| 93 |
-
"std": [
|
| 94 |
-
15.403253151980465,
|
| 95 |
-
43.94546767604488,
|
| 96 |
-
39.75417952413666,
|
| 97 |
-
14.962841885548743,
|
| 98 |
-
2.730520423833518
|
| 99 |
-
],
|
| 100 |
-
"q01": [
|
| 101 |
-
-10.946685590856747,
|
| 102 |
-
-98.72216729939423,
|
| 103 |
-
-9.791353215464346,
|
| 104 |
-
11.800320518831294,
|
| 105 |
-
-53.41379661319665
|
| 106 |
-
],
|
| 107 |
-
"q99": [
|
| 108 |
-
28.462659542451515,
|
| 109 |
-
20.869145111085704,
|
| 110 |
-
99.54463714844687,
|
| 111 |
-
38.68116009046408,
|
| 112 |
-
-49.01597069433932
|
| 113 |
-
]
|
| 114 |
-
},
|
| 115 |
-
"gripper": {
|
| 116 |
-
"max": [
|
| 117 |
-
75.94654846191406
|
| 118 |
-
],
|
| 119 |
-
"min": [
|
| 120 |
-
1.781737208366394
|
| 121 |
-
],
|
| 122 |
-
"mean": [
|
| 123 |
-
15.427439482698658
|
| 124 |
-
],
|
| 125 |
-
"std": [
|
| 126 |
-
13.252202598645871
|
| 127 |
-
],
|
| 128 |
-
"q01": [
|
| 129 |
-
7.224211353540906
|
| 130 |
-
],
|
| 131 |
-
"q99": [
|
| 132 |
-
52.41286561003831
|
| 133 |
-
]
|
| 134 |
-
}
|
| 135 |
-
}
|
| 136 |
-
},
|
| 137 |
-
"modalities": {
|
| 138 |
-
"video": {
|
| 139 |
-
"front": {
|
| 140 |
-
"resolution": [
|
| 141 |
-
640,
|
| 142 |
-
480
|
| 143 |
-
],
|
| 144 |
-
"channels": 3,
|
| 145 |
-
"fps": 30.0
|
| 146 |
-
},
|
| 147 |
-
"wrist": {
|
| 148 |
-
"resolution": [
|
| 149 |
-
640,
|
| 150 |
-
480
|
| 151 |
-
],
|
| 152 |
-
"channels": 3,
|
| 153 |
-
"fps": 30.0
|
| 154 |
-
}
|
| 155 |
-
},
|
| 156 |
-
"state": {
|
| 157 |
-
"single_arm": {
|
| 158 |
-
"absolute": true,
|
| 159 |
-
"rotation_type": null,
|
| 160 |
-
"shape": [
|
| 161 |
-
5
|
| 162 |
-
],
|
| 163 |
-
"continuous": true
|
| 164 |
-
},
|
| 165 |
-
"gripper": {
|
| 166 |
-
"absolute": true,
|
| 167 |
-
"rotation_type": null,
|
| 168 |
-
"shape": [
|
| 169 |
-
1
|
| 170 |
-
],
|
| 171 |
-
"continuous": true
|
| 172 |
-
}
|
| 173 |
-
},
|
| 174 |
-
"action": {
|
| 175 |
-
"single_arm": {
|
| 176 |
-
"absolute": true,
|
| 177 |
-
"rotation_type": null,
|
| 178 |
-
"shape": [
|
| 179 |
-
5
|
| 180 |
-
],
|
| 181 |
-
"continuous": true
|
| 182 |
-
},
|
| 183 |
-
"gripper": {
|
| 184 |
-
"absolute": true,
|
| 185 |
-
"rotation_type": null,
|
| 186 |
-
"shape": [
|
| 187 |
-
1
|
| 188 |
-
],
|
| 189 |
-
"continuous": true
|
| 190 |
-
}
|
| 191 |
-
}
|
| 192 |
-
},
|
| 193 |
-
"embodiment_tag": "new_embodiment"
|
| 194 |
-
}
|
| 195 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/model-00001-of-00002.safetensors
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:ce098cd9a8ddf4ed6040b6ad71d2c472f047a74f76e7834b257d9e0f60ede7b3
|
| 3 |
-
size 4999367032
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/model-00002-of-00002.safetensors
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:64f769fc626d3bfce2db862d7c3db6007eab211c8ca0c90c83fe68111061745a
|
| 3 |
-
size 2586705312
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/model.safetensors.index.json
DELETED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-9000/optimizer.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:5a7fc84fa9c253750e3c2c96fcac87f1d5d726fdc552b3e51c3d99017682fe99
|
| 3 |
-
size 8550720062
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/rng_state.pth
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:474e77e62ecc63ec42f0c8924d50b9029658e0636dd2b968b5016616efa26d82
|
| 3 |
-
size 14244
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/scheduler.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:1c26823aa5685497db4fe82a648513d12cb6c26d9be110887b197c10fe3a44ac
|
| 3 |
-
size 1064
|
|
|
|
|
|
|
|
|
|
|
|
checkpoint-9000/trainer_state.json
DELETED
|
The diff for this file is too large to render.
See raw diff
|
|
|