Upload folder using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- .gitattributes +1 -0
- config.json +80 -0
- experiment_cfg/conf.yaml +0 -0
- experiment_cfg/metadata.json +826 -0
- global_step60000/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_10_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_11_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_12_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_13_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_14_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_15_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_16_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_17_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_18_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_19_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_20_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_21_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_22_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_23_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_24_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_25_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_26_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_27_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_28_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_29_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_30_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_31_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_8_mp_rank_00_optim_states.pt +3 -0
- global_step60000/bf16_zero_pp_rank_9_mp_rank_00_optim_states.pt +3 -0
- global_step60000/mp_rank_00_model_states.pt +3 -0
- latest +1 -0
- model.safetensors +3 -0
- rng_state_0.pth +3 -0
- rng_state_1.pth +3 -0
- rng_state_10.pth +3 -0
- rng_state_11.pth +3 -0
- rng_state_12.pth +3 -0
- rng_state_13.pth +3 -0
- rng_state_14.pth +3 -0
- rng_state_15.pth +3 -0
- rng_state_16.pth +3 -0
- rng_state_17.pth +3 -0
- rng_state_18.pth +3 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
trainer_state.json filter=lfs diff=lfs merge=lfs -text
|
config.json
ADDED
@@ -0,0 +1,80 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "/mnt/amlfs-01/home/seonghyeony/checkpoints/groot_s_idm_24P_300",
|
3 |
+
"action_dim": 32,
|
4 |
+
"action_head_cfg": {
|
5 |
+
"_convert_": "object",
|
6 |
+
"_target_": "gr00t.model.action_head.flow_matching_action_head_idm.FlowMatchingActionHeadIDM",
|
7 |
+
"config": {
|
8 |
+
"_recursive_": false,
|
9 |
+
"_target_": "gr00t.model.action_head.flow_matching_action_head_idm.FlowMatchingActionHeadIDMConfig",
|
10 |
+
"action_dim": 32,
|
11 |
+
"action_horizon": 16,
|
12 |
+
"add_pos_embed": true,
|
13 |
+
"add_seperator_token": true,
|
14 |
+
"add_view_embed": true,
|
15 |
+
"backbone_features_projector_cfg": null,
|
16 |
+
"diffusion_model_cfg": {
|
17 |
+
"_target_": "gr00t.model.action_head.cross_attention_dit.DiT",
|
18 |
+
"attention_head_dim": 64,
|
19 |
+
"dropout": 0.2,
|
20 |
+
"final_dropout": true,
|
21 |
+
"interleave_self_attention": true,
|
22 |
+
"norm_type": "ada_norm",
|
23 |
+
"num_attention_heads": 16,
|
24 |
+
"num_layers": 8,
|
25 |
+
"output_dim": 1024,
|
26 |
+
"positional_embeddings": null
|
27 |
+
},
|
28 |
+
"hidden_size": 1024,
|
29 |
+
"max_action_dim": 32,
|
30 |
+
"max_num_views": 6,
|
31 |
+
"max_state_dim": 44,
|
32 |
+
"mm_projector_cfg": {
|
33 |
+
"_convert_": "object",
|
34 |
+
"_target_": "gr00t.model.action_head.multimodal_projector.MultimodalProjector",
|
35 |
+
"config": {
|
36 |
+
"_target_": "gr00t.model.action_head.multimodal_projector.MultimodalProjectorConfig",
|
37 |
+
"hidden_size": 1024,
|
38 |
+
"mm_hidden_size": 1024,
|
39 |
+
"mm_projector_type": "mlp_doubledownsample"
|
40 |
+
}
|
41 |
+
},
|
42 |
+
"mm_vision_select_layer": -2,
|
43 |
+
"model_dtype": "float32",
|
44 |
+
"noise_beta_alpha": 1.5,
|
45 |
+
"noise_beta_beta": 1.0,
|
46 |
+
"noise_s": 0.999,
|
47 |
+
"num_inference_timesteps": 16,
|
48 |
+
"num_timestep_buckets": 1000,
|
49 |
+
"siglip_hidden_size": 1024,
|
50 |
+
"siglip_model_cfg": {
|
51 |
+
"_convert_": "object",
|
52 |
+
"_target_": "gr00t.model.action_head.siglip.SiglipModel.from_pretrained",
|
53 |
+
"pretrained_model_name_or_path": "google/siglip2-large-patch16-256"
|
54 |
+
},
|
55 |
+
"tune_vision_tower": true,
|
56 |
+
"vl_self_attention_cfg": {
|
57 |
+
"_target_": "gr00t.model.action_head.cross_attention_dit.SelfAttentionTransformer",
|
58 |
+
"attention_head_dim": 64,
|
59 |
+
"dropout": 0.2,
|
60 |
+
"final_dropout": true,
|
61 |
+
"num_attention_heads": 16,
|
62 |
+
"num_layers": 4,
|
63 |
+
"positional_embeddings": null
|
64 |
+
}
|
65 |
+
}
|
66 |
+
},
|
67 |
+
"action_horizon": 16,
|
68 |
+
"architectures": [
|
69 |
+
"DualBrain"
|
70 |
+
],
|
71 |
+
"backbone_cfg": {
|
72 |
+
"_target_": "gr00t.model.backbone.IdentityBackbone"
|
73 |
+
},
|
74 |
+
"hidden_size": 0,
|
75 |
+
"model_dtype": "float32",
|
76 |
+
"model_type": "dual_brain",
|
77 |
+
"resume_path": "/mnt/amlfs-01/home/seonghyeony/checkpoints/groot_s_idm_24P_300",
|
78 |
+
"torch_dtype": "bfloat16",
|
79 |
+
"transformers_version": "4.45.2"
|
80 |
+
}
|
experiment_cfg/conf.yaml
ADDED
The diff for this file is too large to render.
See raw diff
|
|
experiment_cfg/metadata.json
ADDED
@@ -0,0 +1,826 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"robocasa_panda_omron": {
|
3 |
+
"dataset_name": "robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet, robocasa_panda_omron:FullSet",
|
4 |
+
"dataset_statistics": {
|
5 |
+
"state": {
|
6 |
+
"base_position": {
|
7 |
+
"max": [
|
8 |
+
5.60412073135376,
|
9 |
+
-0.4872598946094513,
|
10 |
+
0.8319682478904724
|
11 |
+
],
|
12 |
+
"min": [
|
13 |
+
0.31250813603401184,
|
14 |
+
-5.110836982727051,
|
15 |
+
0.6947912573814392
|
16 |
+
],
|
17 |
+
"mean": [
|
18 |
+
2.5233612060546875,
|
19 |
+
-1.4497382640838627,
|
20 |
+
0.7089967727661134
|
21 |
+
],
|
22 |
+
"std": [
|
23 |
+
1.4186391830444336,
|
24 |
+
0.8250066041946406,
|
25 |
+
0.008732615038746
|
26 |
+
],
|
27 |
+
"q01": [
|
28 |
+
0.5000049471855164,
|
29 |
+
-4.100456237792969,
|
30 |
+
0.6999498009681703
|
31 |
+
],
|
32 |
+
"q99": [
|
33 |
+
5.5999813079834,
|
34 |
+
-0.7786697745323181,
|
35 |
+
0.7069445848464966
|
36 |
+
]
|
37 |
+
},
|
38 |
+
"base_rotation": {
|
39 |
+
"max": [
|
40 |
+
0.0,
|
41 |
+
0.0,
|
42 |
+
1.0,
|
43 |
+
1.0
|
44 |
+
],
|
45 |
+
"min": [
|
46 |
+
0.0,
|
47 |
+
0.0,
|
48 |
+
-1.0,
|
49 |
+
0.0
|
50 |
+
],
|
51 |
+
"mean": [
|
52 |
+
0.0,
|
53 |
+
0.0,
|
54 |
+
0.25808697938919073,
|
55 |
+
0.6459704637527465
|
56 |
+
],
|
57 |
+
"std": [
|
58 |
+
0.0,
|
59 |
+
0.0,
|
60 |
+
0.5877522826194763,
|
61 |
+
0.3639818727970123
|
62 |
+
],
|
63 |
+
"q01": [
|
64 |
+
0.0,
|
65 |
+
0.0,
|
66 |
+
-0.9999999999999999,
|
67 |
+
5.5698886990285226e-06
|
68 |
+
],
|
69 |
+
"q99": [
|
70 |
+
0.0,
|
71 |
+
0.0,
|
72 |
+
0.9999999999999999,
|
73 |
+
0.9999999999999999
|
74 |
+
]
|
75 |
+
},
|
76 |
+
"end_effector_position_absolute": {
|
77 |
+
"max": [
|
78 |
+
6.200742721557617,
|
79 |
+
-0.00844558235257864,
|
80 |
+
1.8568346500396729
|
81 |
+
],
|
82 |
+
"min": [
|
83 |
+
0.011350978165864944,
|
84 |
+
-5.129131317138672,
|
85 |
+
0.6468141078948975
|
86 |
+
],
|
87 |
+
"mean": [
|
88 |
+
2.5114300251007076,
|
89 |
+
-1.360061168670654,
|
90 |
+
1.339444875717163
|
91 |
+
],
|
92 |
+
"std": [
|
93 |
+
1.5792435407638563,
|
94 |
+
0.9657800793647767,
|
95 |
+
0.25533461570739835
|
96 |
+
],
|
97 |
+
"q01": [
|
98 |
+
0.15726516664028167,
|
99 |
+
-4.127151670455933,
|
100 |
+
0.7751663839817047
|
101 |
+
],
|
102 |
+
"q99": [
|
103 |
+
5.854645910263061,
|
104 |
+
-0.14227030426263923,
|
105 |
+
1.6728234744071961
|
106 |
+
]
|
107 |
+
},
|
108 |
+
"end_effector_position_relative": {
|
109 |
+
"max": [
|
110 |
+
0.8734620809555054,
|
111 |
+
0.9110966920852661,
|
112 |
+
1.1568338871002197
|
113 |
+
],
|
114 |
+
"min": [
|
115 |
+
-0.46594083309173584,
|
116 |
+
-0.8583298325538635,
|
117 |
+
-0.05373392626643181
|
118 |
+
],
|
119 |
+
"mean": [
|
120 |
+
0.2642126679420471,
|
121 |
+
-0.037328604608774185,
|
122 |
+
0.5461296439170839
|
123 |
+
],
|
124 |
+
"std": [
|
125 |
+
0.18807946145534513,
|
126 |
+
0.2574712634086609,
|
127 |
+
0.23820365965366327
|
128 |
+
],
|
129 |
+
"q01": [
|
130 |
+
-0.25546322107315067,
|
131 |
+
-0.6193294751644135,
|
132 |
+
0.07401651039719581
|
133 |
+
],
|
134 |
+
"q99": [
|
135 |
+
0.7370126557350158,
|
136 |
+
0.6061198198795318,
|
137 |
+
0.9726848316192626
|
138 |
+
]
|
139 |
+
},
|
140 |
+
"end_effector_rotation_absolute": {
|
141 |
+
"max": [
|
142 |
+
0.9999977350234985,
|
143 |
+
0.9999508857727051,
|
144 |
+
0.9938080310821533,
|
145 |
+
0.9647020697593689
|
146 |
+
],
|
147 |
+
"min": [
|
148 |
+
-0.9724215269088745,
|
149 |
+
-0.9999970197677612,
|
150 |
+
-0.9864645600318909,
|
151 |
+
-0.9823317527770996
|
152 |
+
],
|
153 |
+
"mean": [
|
154 |
+
0.47370263934135426,
|
155 |
+
0.0261838398873806,
|
156 |
+
0.16200047731399536,
|
157 |
+
-0.04991654306650162
|
158 |
+
],
|
159 |
+
"std": [
|
160 |
+
0.41681185364723217,
|
161 |
+
0.608587920665741,
|
162 |
+
0.2949107885360718,
|
163 |
+
0.3205944299697876
|
164 |
+
],
|
165 |
+
"q01": [
|
166 |
+
-0.6978112339973449,
|
167 |
+
-0.9920166134834288,
|
168 |
+
-0.6591729152202607,
|
169 |
+
-0.7654285311698916
|
170 |
+
],
|
171 |
+
"q99": [
|
172 |
+
0.9918085932731626,
|
173 |
+
0.9379134571552272,
|
174 |
+
0.7881702184677124,
|
175 |
+
0.7439913916587824
|
176 |
+
]
|
177 |
+
},
|
178 |
+
"end_effector_rotation_relative": {
|
179 |
+
"max": [
|
180 |
+
0.9999980926513672,
|
181 |
+
0.998063862323761,
|
182 |
+
0.9959006905555725,
|
183 |
+
0.9754553437232971
|
184 |
+
],
|
185 |
+
"min": [
|
186 |
+
-0.9999996423721313,
|
187 |
+
-0.9978545904159546,
|
188 |
+
-0.9934196472167969,
|
189 |
+
1.742676403182486e-08
|
190 |
+
],
|
191 |
+
"mean": [
|
192 |
+
-0.15737581253051755,
|
193 |
+
0.12279410660266878,
|
194 |
+
0.0075049446895718575,
|
195 |
+
0.19545103609561915
|
196 |
+
],
|
197 |
+
"std": [
|
198 |
+
0.7716760039329528,
|
199 |
+
0.35558718442916865,
|
200 |
+
0.34857386350631714,
|
201 |
+
0.23019900918006903
|
202 |
+
],
|
203 |
+
"q01": [
|
204 |
+
-0.9974887371063231,
|
205 |
+
-0.6873972046375275,
|
206 |
+
-0.774531364440918,
|
207 |
+
0.0013910925900563597
|
208 |
+
],
|
209 |
+
"q99": [
|
210 |
+
0.9955765008926392,
|
211 |
+
0.8495244240760801,
|
212 |
+
0.7640593051910399,
|
213 |
+
0.8050539493560789
|
214 |
+
]
|
215 |
+
},
|
216 |
+
"gripper_qpos": {
|
217 |
+
"max": [
|
218 |
+
0.0993952602148056,
|
219 |
+
0.12181679159402847
|
220 |
+
],
|
221 |
+
"min": [
|
222 |
+
-0.034708939492702484,
|
223 |
+
-0.12887509167194366
|
224 |
+
],
|
225 |
+
"mean": [
|
226 |
+
0.03116090223193169,
|
227 |
+
-0.0305799338966608
|
228 |
+
],
|
229 |
+
"std": [
|
230 |
+
0.01336925104260443,
|
231 |
+
0.01332307420670987
|
232 |
+
],
|
233 |
+
"q01": [
|
234 |
+
-4.0376453907811085e-05,
|
235 |
+
-0.040308743715286255
|
236 |
+
],
|
237 |
+
"q99": [
|
238 |
+
0.04063318811357021,
|
239 |
+
-0.00040857097716070936
|
240 |
+
]
|
241 |
+
},
|
242 |
+
"gripper_qvel": {
|
243 |
+
"max": [
|
244 |
+
3.6628172397613525,
|
245 |
+
16.347944259643555
|
246 |
+
],
|
247 |
+
"min": [
|
248 |
+
-3.065218210220337,
|
249 |
+
-1.0290303230285645
|
250 |
+
],
|
251 |
+
"mean": [
|
252 |
+
0.0007271775975823402,
|
253 |
+
-0.0007187921437434852
|
254 |
+
],
|
255 |
+
"std": [
|
256 |
+
0.01484494563192129,
|
257 |
+
0.01693008281290531
|
258 |
+
],
|
259 |
+
"q01": [
|
260 |
+
-0.06051826104521752,
|
261 |
+
-0.06142193399369715
|
262 |
+
],
|
263 |
+
"q99": [
|
264 |
+
0.06037045575678332,
|
265 |
+
0.06032651640474791
|
266 |
+
]
|
267 |
+
},
|
268 |
+
"joint_position": {
|
269 |
+
"max": [
|
270 |
+
2.914743423461914,
|
271 |
+
1.7686904668807983,
|
272 |
+
2.916494846343994,
|
273 |
+
-0.016307059675455093,
|
274 |
+
2.939373016357422,
|
275 |
+
3.790846109390259,
|
276 |
+
2.954829454421997
|
277 |
+
],
|
278 |
+
"min": [
|
279 |
+
-2.907266616821289,
|
280 |
+
-1.7821346521377563,
|
281 |
+
-2.9019548892974854,
|
282 |
+
-3.095099449157715,
|
283 |
+
-2.937913417816162,
|
284 |
+
-0.0009604570223018527,
|
285 |
+
-2.92836594581604
|
286 |
+
],
|
287 |
+
"mean": [
|
288 |
+
-0.002226361073553562,
|
289 |
+
-0.7944630384445189,
|
290 |
+
-0.13673430681228638,
|
291 |
+
-2.1579504013061523,
|
292 |
+
0.22298482060432434,
|
293 |
+
1.9989188909530637,
|
294 |
+
0.6232013106346129
|
295 |
+
],
|
296 |
+
"std": [
|
297 |
+
0.39000499248504633,
|
298 |
+
0.6092476248741152,
|
299 |
+
0.6991883516311646,
|
300 |
+
0.6551395654678345,
|
301 |
+
0.736649513244629,
|
302 |
+
0.568905234336853,
|
303 |
+
0.8584070205688478
|
304 |
+
],
|
305 |
+
"q01": [
|
306 |
+
-1.1111454963684084,
|
307 |
+
-1.7629053592681887,
|
308 |
+
-1.9500407242774964,
|
309 |
+
-3.0719740390777583,
|
310 |
+
-1.3464225411415103,
|
311 |
+
0.710475721359253,
|
312 |
+
-1.6062000060081483
|
313 |
+
],
|
314 |
+
"q99": [
|
315 |
+
1.4440935993194488,
|
316 |
+
0.7535663640499111,
|
317 |
+
1.9061625623702998,
|
318 |
+
-0.06965120136737823,
|
319 |
+
2.8974728584289555,
|
320 |
+
3.515558786392211,
|
321 |
+
2.689405360221862
|
322 |
+
]
|
323 |
+
},
|
324 |
+
"joint_position_cos": {
|
325 |
+
"max": [
|
326 |
+
1.0,
|
327 |
+
1.0,
|
328 |
+
1.0,
|
329 |
+
0.9998670220375061,
|
330 |
+
1.0,
|
331 |
+
0.9999995231628418,
|
332 |
+
1.0
|
333 |
+
],
|
334 |
+
"min": [
|
335 |
+
-0.9743798971176147,
|
336 |
+
-0.20976868271827698,
|
337 |
+
-0.9747722744941711,
|
338 |
+
-0.9989193677902222,
|
339 |
+
-0.9796231389045715,
|
340 |
+
-1.0,
|
341 |
+
-0.9826104044914246
|
342 |
+
],
|
343 |
+
"mean": [
|
344 |
+
0.8548306226730348,
|
345 |
+
0.5775746703147887,
|
346 |
+
0.8072980046272277,
|
347 |
+
-0.4509732127189635,
|
348 |
+
0.8170415759086608,
|
349 |
+
-0.32237827777862554,
|
350 |
+
0.5677806138992308
|
351 |
+
],
|
352 |
+
"std": [
|
353 |
+
0.18712373077869357,
|
354 |
+
0.3143303990364076,
|
355 |
+
0.3379428088665012,
|
356 |
+
0.5030199885368347,
|
357 |
+
0.4237685203552249,
|
358 |
+
0.4407167136669159,
|
359 |
+
0.44018274545669567
|
360 |
+
],
|
361 |
+
"q01": [
|
362 |
+
-0.02635294888168575,
|
363 |
+
-0.19092965126037595,
|
364 |
+
-0.431967819929123,
|
365 |
+
-0.9975776076316832,
|
366 |
+
-0.9703514575958252,
|
367 |
+
-0.9988470673561097,
|
368 |
+
-0.927694798707962
|
369 |
+
],
|
370 |
+
"q99": [
|
371 |
+
0.9999993443489074,
|
372 |
+
0.9995868206024171,
|
373 |
+
0.9999943375587463,
|
374 |
+
0.9975753426551818,
|
375 |
+
0.9999939799308778,
|
376 |
+
0.7580517125129691,
|
377 |
+
0.9997406005859374
|
378 |
+
]
|
379 |
+
},
|
380 |
+
"joint_position_sin": {
|
381 |
+
"max": [
|
382 |
+
1.0,
|
383 |
+
1.0,
|
384 |
+
1.0,
|
385 |
+
-0.016306336969137192,
|
386 |
+
1.0,
|
387 |
+
1.0,
|
388 |
+
1.0
|
389 |
+
],
|
390 |
+
"min": [
|
391 |
+
-1.0,
|
392 |
+
-1.0,
|
393 |
+
-1.0,
|
394 |
+
-1.0,
|
395 |
+
-1.0,
|
396 |
+
-0.6045919060707092,
|
397 |
+
-1.0
|
398 |
+
],
|
399 |
+
"mean": [
|
400 |
+
-0.009903629310429098,
|
401 |
+
-0.6532469391822815,
|
402 |
+
-0.11583427339792252,
|
403 |
+
-0.7166314125061036,
|
404 |
+
0.10584983974695206,
|
405 |
+
0.8206916451454164,
|
406 |
+
0.42647859454154957
|
407 |
+
],
|
408 |
+
"std": [
|
409 |
+
0.30439406633377075,
|
410 |
+
0.45045891404151917,
|
411 |
+
0.49071264266967773,
|
412 |
+
0.26929175853729226,
|
413 |
+
0.41677382588386536,
|
414 |
+
0.28926900029182395,
|
415 |
+
0.5585145950317384
|
416 |
+
],
|
417 |
+
"q01": [
|
418 |
+
-0.8871335268020629,
|
419 |
+
-0.9998832941055298,
|
420 |
+
-0.9991828799247741,
|
421 |
+
-0.9997999668121336,
|
422 |
+
-0.9557854759693147,
|
423 |
+
-0.3653103184700012,
|
424 |
+
-0.9881189262866972
|
425 |
+
],
|
426 |
+
"q99": [
|
427 |
+
0.9552828037738794,
|
428 |
+
0.6842438745498653,
|
429 |
+
0.997663722038269,
|
430 |
+
-0.06941956743597984,
|
431 |
+
0.9985226392745972,
|
432 |
+
0.9999756217002866,
|
433 |
+
0.9997976422309875
|
434 |
+
]
|
435 |
+
},
|
436 |
+
"joint_velocity": {
|
437 |
+
"max": [
|
438 |
+
4.507856369018555,
|
439 |
+
4.772953987121582,
|
440 |
+
5.447414875030518,
|
441 |
+
6.97592830657959,
|
442 |
+
11.213868141174316,
|
443 |
+
8.45352554321289,
|
444 |
+
7.553228378295898
|
445 |
+
],
|
446 |
+
"min": [
|
447 |
+
-4.643657207489014,
|
448 |
+
-3.8053159713745117,
|
449 |
+
-10.225778579711914,
|
450 |
+
-5.300220489501953,
|
451 |
+
-8.159320831298828,
|
452 |
+
-7.285642147064209,
|
453 |
+
-6.582308292388916
|
454 |
+
],
|
455 |
+
"mean": [
|
456 |
+
-0.0051504699513316154,
|
457 |
+
0.017983481287956238,
|
458 |
+
-0.0032231842633336795,
|
459 |
+
0.018490538001060486,
|
460 |
+
0.01154279615730047,
|
461 |
+
0.03427882865071296,
|
462 |
+
-0.008020073175430298
|
463 |
+
],
|
464 |
+
"std": [
|
465 |
+
0.22057239711284635,
|
466 |
+
0.3303338885307312,
|
467 |
+
0.25083741545677185,
|
468 |
+
0.3530645966529846,
|
469 |
+
0.3323102593421936,
|
470 |
+
0.3750852048397064,
|
471 |
+
0.3499438166618347
|
472 |
+
],
|
473 |
+
"q01": [
|
474 |
+
-0.736943724155426,
|
475 |
+
-1.0061864113807681,
|
476 |
+
-0.6962891066074371,
|
477 |
+
-1.0777667713165284,
|
478 |
+
-0.9365327370166779,
|
479 |
+
-1.0524059319496155,
|
480 |
+
-1.0341050219535823
|
481 |
+
],
|
482 |
+
"q99": [
|
483 |
+
0.6565076839923846,
|
484 |
+
0.8276887702941891,
|
485 |
+
0.7165593433380111,
|
486 |
+
1.0437071704864485,
|
487 |
+
1.0606284761428828,
|
488 |
+
1.047166392803192,
|
489 |
+
0.9755567121505715
|
490 |
+
]
|
491 |
+
}
|
492 |
+
},
|
493 |
+
"action": {
|
494 |
+
"base_motion": {
|
495 |
+
"max": [
|
496 |
+
0.0,
|
497 |
+
0.0,
|
498 |
+
0.0,
|
499 |
+
0.0
|
500 |
+
],
|
501 |
+
"min": [
|
502 |
+
0.0,
|
503 |
+
0.0,
|
504 |
+
0.0,
|
505 |
+
0.0
|
506 |
+
],
|
507 |
+
"mean": [
|
508 |
+
0.0,
|
509 |
+
0.0,
|
510 |
+
0.0,
|
511 |
+
0.0
|
512 |
+
],
|
513 |
+
"std": [
|
514 |
+
0.0,
|
515 |
+
0.0,
|
516 |
+
0.0,
|
517 |
+
0.0
|
518 |
+
],
|
519 |
+
"q01": [
|
520 |
+
0.0,
|
521 |
+
0.0,
|
522 |
+
0.0,
|
523 |
+
0.0
|
524 |
+
],
|
525 |
+
"q99": [
|
526 |
+
0.0,
|
527 |
+
0.0,
|
528 |
+
0.0,
|
529 |
+
0.0
|
530 |
+
]
|
531 |
+
},
|
532 |
+
"control_mode": {
|
533 |
+
"max": [
|
534 |
+
0.0
|
535 |
+
],
|
536 |
+
"min": [
|
537 |
+
0.0
|
538 |
+
],
|
539 |
+
"mean": [
|
540 |
+
0.0
|
541 |
+
],
|
542 |
+
"std": [
|
543 |
+
0.0
|
544 |
+
],
|
545 |
+
"q01": [
|
546 |
+
0.0
|
547 |
+
],
|
548 |
+
"q99": [
|
549 |
+
0.0
|
550 |
+
]
|
551 |
+
},
|
552 |
+
"end_effector_position": {
|
553 |
+
"max": [
|
554 |
+
1.0,
|
555 |
+
1.0,
|
556 |
+
1.0
|
557 |
+
],
|
558 |
+
"min": [
|
559 |
+
-1.0,
|
560 |
+
-1.0,
|
561 |
+
-1.0
|
562 |
+
],
|
563 |
+
"mean": [
|
564 |
+
0.03037301078438759,
|
565 |
+
0.003446326823905111,
|
566 |
+
-0.040158476680517204
|
567 |
+
],
|
568 |
+
"std": [
|
569 |
+
0.4068017899990082,
|
570 |
+
0.3985259532928467,
|
571 |
+
0.36361521482467646
|
572 |
+
],
|
573 |
+
"q01": [
|
574 |
+
-0.9999999999999999,
|
575 |
+
-0.9999999999999999,
|
576 |
+
-0.9999999999999999
|
577 |
+
],
|
578 |
+
"q99": [
|
579 |
+
0.9999999999999999,
|
580 |
+
0.9999999999999999,
|
581 |
+
0.9999999999999999
|
582 |
+
]
|
583 |
+
},
|
584 |
+
"end_effector_rotation": {
|
585 |
+
"max": [
|
586 |
+
1.0,
|
587 |
+
1.0,
|
588 |
+
1.0
|
589 |
+
],
|
590 |
+
"min": [
|
591 |
+
-1.0,
|
592 |
+
-1.0,
|
593 |
+
-1.0
|
594 |
+
],
|
595 |
+
"mean": [
|
596 |
+
0.006670467555522919,
|
597 |
+
-0.022744668647646907,
|
598 |
+
-0.002840776927769184
|
599 |
+
],
|
600 |
+
"std": [
|
601 |
+
0.14142614603042605,
|
602 |
+
0.1476612389087677,
|
603 |
+
0.13538147509098053
|
604 |
+
],
|
605 |
+
"q01": [
|
606 |
+
-0.41466328322887414,
|
607 |
+
-0.508793534040451,
|
608 |
+
-0.4084805303812027
|
609 |
+
],
|
610 |
+
"q99": [
|
611 |
+
0.4669318711757646,
|
612 |
+
0.3791767048835726,
|
613 |
+
0.4212867003679271
|
614 |
+
]
|
615 |
+
},
|
616 |
+
"gripper_close": {
|
617 |
+
"max": [
|
618 |
+
1.0
|
619 |
+
],
|
620 |
+
"min": [
|
621 |
+
0.0
|
622 |
+
],
|
623 |
+
"mean": [
|
624 |
+
0.3556057214736939
|
625 |
+
],
|
626 |
+
"std": [
|
627 |
+
0.49354264140129084
|
628 |
+
],
|
629 |
+
"q01": [
|
630 |
+
0.0
|
631 |
+
],
|
632 |
+
"q99": [
|
633 |
+
1.0
|
634 |
+
]
|
635 |
+
}
|
636 |
+
},
|
637 |
+
"total_trajectory_length": 9288,
|
638 |
+
"num_trajectories": 1691280
|
639 |
+
},
|
640 |
+
"modalities": {
|
641 |
+
"video": {
|
642 |
+
"res256_image_side_0": {
|
643 |
+
"resolution": [
|
644 |
+
256,
|
645 |
+
256
|
646 |
+
],
|
647 |
+
"channels": 3,
|
648 |
+
"fps": 20.0
|
649 |
+
},
|
650 |
+
"res256_image_side_1": {
|
651 |
+
"resolution": [
|
652 |
+
256,
|
653 |
+
256
|
654 |
+
],
|
655 |
+
"channels": 3,
|
656 |
+
"fps": 20.0
|
657 |
+
},
|
658 |
+
"res256_image_wrist_0": {
|
659 |
+
"resolution": [
|
660 |
+
256,
|
661 |
+
256
|
662 |
+
],
|
663 |
+
"channels": 3,
|
664 |
+
"fps": 20.0
|
665 |
+
}
|
666 |
+
},
|
667 |
+
"state": {
|
668 |
+
"base_position": {
|
669 |
+
"absolute": true,
|
670 |
+
"rotation_type": null,
|
671 |
+
"shape": [
|
672 |
+
3
|
673 |
+
],
|
674 |
+
"continuous": true
|
675 |
+
},
|
676 |
+
"base_rotation": {
|
677 |
+
"absolute": true,
|
678 |
+
"rotation_type": "quaternion",
|
679 |
+
"shape": [
|
680 |
+
4
|
681 |
+
],
|
682 |
+
"continuous": true
|
683 |
+
},
|
684 |
+
"end_effector_position_absolute": {
|
685 |
+
"absolute": true,
|
686 |
+
"rotation_type": null,
|
687 |
+
"shape": [
|
688 |
+
3
|
689 |
+
],
|
690 |
+
"continuous": true
|
691 |
+
},
|
692 |
+
"end_effector_position_relative": {
|
693 |
+
"absolute": true,
|
694 |
+
"rotation_type": null,
|
695 |
+
"shape": [
|
696 |
+
3
|
697 |
+
],
|
698 |
+
"continuous": true
|
699 |
+
},
|
700 |
+
"end_effector_rotation_absolute": {
|
701 |
+
"absolute": true,
|
702 |
+
"rotation_type": "quaternion",
|
703 |
+
"shape": [
|
704 |
+
4
|
705 |
+
],
|
706 |
+
"continuous": true
|
707 |
+
},
|
708 |
+
"end_effector_rotation_relative": {
|
709 |
+
"absolute": true,
|
710 |
+
"rotation_type": "quaternion",
|
711 |
+
"shape": [
|
712 |
+
4
|
713 |
+
],
|
714 |
+
"continuous": true
|
715 |
+
},
|
716 |
+
"gripper_qpos": {
|
717 |
+
"absolute": true,
|
718 |
+
"rotation_type": null,
|
719 |
+
"shape": [
|
720 |
+
2
|
721 |
+
],
|
722 |
+
"continuous": true
|
723 |
+
},
|
724 |
+
"gripper_qvel": {
|
725 |
+
"absolute": true,
|
726 |
+
"rotation_type": null,
|
727 |
+
"shape": [
|
728 |
+
2
|
729 |
+
],
|
730 |
+
"continuous": true
|
731 |
+
},
|
732 |
+
"joint_position": {
|
733 |
+
"absolute": true,
|
734 |
+
"rotation_type": null,
|
735 |
+
"shape": [
|
736 |
+
7
|
737 |
+
],
|
738 |
+
"continuous": true
|
739 |
+
},
|
740 |
+
"joint_position_cos": {
|
741 |
+
"absolute": true,
|
742 |
+
"rotation_type": null,
|
743 |
+
"shape": [
|
744 |
+
7
|
745 |
+
],
|
746 |
+
"continuous": true
|
747 |
+
},
|
748 |
+
"joint_position_sin": {
|
749 |
+
"absolute": true,
|
750 |
+
"rotation_type": null,
|
751 |
+
"shape": [
|
752 |
+
7
|
753 |
+
],
|
754 |
+
"continuous": true
|
755 |
+
},
|
756 |
+
"joint_velocity": {
|
757 |
+
"absolute": true,
|
758 |
+
"rotation_type": null,
|
759 |
+
"shape": [
|
760 |
+
7
|
761 |
+
],
|
762 |
+
"continuous": true
|
763 |
+
}
|
764 |
+
},
|
765 |
+
"action": {
|
766 |
+
"base_motion": {
|
767 |
+
"absolute": false,
|
768 |
+
"rotation_type": null,
|
769 |
+
"shape": [
|
770 |
+
4
|
771 |
+
],
|
772 |
+
"continuous": true
|
773 |
+
},
|
774 |
+
"control_mode": {
|
775 |
+
"absolute": true,
|
776 |
+
"rotation_type": null,
|
777 |
+
"shape": [
|
778 |
+
1
|
779 |
+
],
|
780 |
+
"continuous": false
|
781 |
+
},
|
782 |
+
"end_effector_position": {
|
783 |
+
"absolute": false,
|
784 |
+
"rotation_type": null,
|
785 |
+
"shape": [
|
786 |
+
3
|
787 |
+
],
|
788 |
+
"continuous": true
|
789 |
+
},
|
790 |
+
"end_effector_rotation": {
|
791 |
+
"absolute": false,
|
792 |
+
"rotation_type": "axis_angle",
|
793 |
+
"shape": [
|
794 |
+
3
|
795 |
+
],
|
796 |
+
"continuous": true
|
797 |
+
},
|
798 |
+
"gripper_close": {
|
799 |
+
"absolute": true,
|
800 |
+
"rotation_type": null,
|
801 |
+
"shape": [
|
802 |
+
1
|
803 |
+
],
|
804 |
+
"continuous": false
|
805 |
+
}
|
806 |
+
},
|
807 |
+
"annotation": {
|
808 |
+
"human": [
|
809 |
+
"action.task_description",
|
810 |
+
"action.task_name",
|
811 |
+
"validity"
|
812 |
+
]
|
813 |
+
}
|
814 |
+
},
|
815 |
+
"embodiment": {
|
816 |
+
"robot_name": "Franka",
|
817 |
+
"robot_type": "Panda+OmronBase",
|
818 |
+
"record_frequency": 20.0,
|
819 |
+
"body_controller_frequency": null,
|
820 |
+
"hand_controller_frequency": null,
|
821 |
+
"embodiment_tag": "robocasa_panda_omron"
|
822 |
+
},
|
823 |
+
"processing": null,
|
824 |
+
"version": null
|
825 |
+
}
|
826 |
+
}
|
global_step60000/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:61a7dbb2d251044deb8f9acff69d9a8797ec150120943c7286831021e6cd344f
|
3 |
+
size 229582448
|
global_step60000/bf16_zero_pp_rank_10_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b31d4b429a8a14d5ce8e424f18eaf160610d8f75f9f459d54e3e3dd69ae404c
|
3 |
+
size 229582140
|
global_step60000/bf16_zero_pp_rank_11_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ddc1012d3521da3a017aa15d374259c4abdb2f0d52b830cd8eb67f3b77657ec0
|
3 |
+
size 229582268
|
global_step60000/bf16_zero_pp_rank_12_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e8f0dcef5cb996c826afb6a3de48a412bd7bb106c20bbec14364478e69696c34
|
3 |
+
size 229582140
|
global_step60000/bf16_zero_pp_rank_13_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5f808d2290a2328aad0bb97c6149c3dc2778ff2308bf05e2badcb09c93996697
|
3 |
+
size 229582140
|
global_step60000/bf16_zero_pp_rank_14_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ae17b7810ccfa4f7aaf4521645cb604c69a5752213823bd5e59a58a3ee8f9e2d
|
3 |
+
size 229582140
|
global_step60000/bf16_zero_pp_rank_15_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:21ca9fca0ba43394b6c2fa6735195897163250f1627856a6de0a522aeab5165d
|
3 |
+
size 229582076
|
global_step60000/bf16_zero_pp_rank_16_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d95dab1fb53717cd9cfbc147459cb13ef57197be641d3e46a9d9f64f0380bc17
|
3 |
+
size 229581948
|
global_step60000/bf16_zero_pp_rank_17_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:12677d2ca71d20331c5793349c9d4497cab4a4fbb0e443b87b111689413082d1
|
3 |
+
size 229582268
|
global_step60000/bf16_zero_pp_rank_18_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dce6ffab5363b1c6b8cf870e3dfb0a93913e4703687872658754401d1232fec2
|
3 |
+
size 229581884
|
global_step60000/bf16_zero_pp_rank_19_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:22bfc4d32d70da39933a5161d4c02ff6bdd222adfe9f53a3db60dd6b3c498cdc
|
3 |
+
size 229581948
|
global_step60000/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a0beaaf9b55b900d8e660eaac3c510098962a2d9500915f2bc611a188ceacd20
|
3 |
+
size 229582000
|
global_step60000/bf16_zero_pp_rank_20_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b148fb577fa3c9f7100dc0f08de4a4f5f8a83d0f25c332501b21b37c3567c7c
|
3 |
+
size 229582012
|
global_step60000/bf16_zero_pp_rank_21_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3697a20e7e6020eb408f9955d74757d6f940f998906ab311fb94d6bbf773ee6f
|
3 |
+
size 229581692
|
global_step60000/bf16_zero_pp_rank_22_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9320cd5dd46980821ccb5871d9719cb3a9b8b2bcfeba281890bf62fb19af0702
|
3 |
+
size 229581692
|
global_step60000/bf16_zero_pp_rank_23_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:824cd3b2fe93cde3e90673d5eb76292238f217eb5d06ed86f18c54ccc172aed2
|
3 |
+
size 229581756
|
global_step60000/bf16_zero_pp_rank_24_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bd137aca3e137da84d345e38bebdd37f5d701a424489a3ad06fb2fb875827ea7
|
3 |
+
size 229581180
|
global_step60000/bf16_zero_pp_rank_25_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b3d159b12a4466a0cb31f19b06386291e40cfa26ddbb12b715c5b4d87e96eb0f
|
3 |
+
size 229580668
|
global_step60000/bf16_zero_pp_rank_26_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:04119e882238ae560480e38eb3c57c42298a9ae9ca480e1b579c2a3a52244cca
|
3 |
+
size 229580732
|
global_step60000/bf16_zero_pp_rank_27_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5a27132dae2b11fb83822ba08111d1a601bf386a2229668388cba8d0a18c7e9c
|
3 |
+
size 229581244
|
global_step60000/bf16_zero_pp_rank_28_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e1eafc89c95c3bcc9f2e91770e68fad9ac8f190cd15b5ca37b981c9be10e5a4e
|
3 |
+
size 229581180
|
global_step60000/bf16_zero_pp_rank_29_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:35d74c93993be53ad46a1f7f95c11096335833852fceb1e9b561bdb21fcce28e
|
3 |
+
size 229581244
|
global_step60000/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3397fac9272c3247df75b04ba6d4c39c2d0805aa037bf8acca4098c747a96b24
|
3 |
+
size 229582320
|
global_step60000/bf16_zero_pp_rank_30_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:32e0b1c19f6cdb9765abdc7c4b77bf4e3de99060c1fabc30b62719afd942c7a5
|
3 |
+
size 229580732
|
global_step60000/bf16_zero_pp_rank_31_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0118cd25a916783426663e72fe8697e3b084c8c03757a8f0a4f3b85d2900cf46
|
3 |
+
size 229580988
|
global_step60000/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a26201057835f03cf2570c52770d5dc299909f6242e26b5a8af5386964e25cf1
|
3 |
+
size 229582000
|
global_step60000/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7eb66ab81a30e35ea7ae512dd4fb9d36615ca1ef4ffb5cf128d3eb662a6568ee
|
3 |
+
size 229582320
|
global_step60000/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:40b17016e84e25b047fea4fe88223163f7f2a82a5fde149fad1283c66d09233d
|
3 |
+
size 229581872
|
global_step60000/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:98a052411719c96070e3ffb4c42d20375df2eb1da87af4664c9387cf77815d3e
|
3 |
+
size 229582256
|
global_step60000/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:30d9ab0ad83d200ffc89a41122762d48e52c29d4501140b5a096fc2eb72c965d
|
3 |
+
size 229582000
|
global_step60000/bf16_zero_pp_rank_8_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6e0879971bbb1a7ac429e144e49e3c1bf67fb92c441309612f6e609fab6d6e40
|
3 |
+
size 229582256
|
global_step60000/bf16_zero_pp_rank_9_mp_rank_00_optim_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b986897bd9748bd2ea23b3a9dd786e0ad2029e34a2ae5e4e1db5131fa1a24a75
|
3 |
+
size 229582256
|
global_step60000/mp_rank_00_model_states.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e865663cf02bb399c9c3c2b5b733ac95c9b16e8a890640bc6ff3ebeede071bd3
|
3 |
+
size 1325393186
|
latest
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
global_step60000
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5f30bcb0b53eba6e4225fec41e8b953c7cd77221320aaca1285e3e5321cc3b8e
|
3 |
+
size 1274869220
|
rng_state_0.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b5a06ff249b9991f1fd615d938bda8244b255458c851656ac52db6479ce1940
|
3 |
+
size 14256
|
rng_state_1.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:94d40bdbddd3cd698c19a0494bda3b649ed693b95ca1d680a10f37b2eccf250b
|
3 |
+
size 14256
|
rng_state_10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a33f183849df8388d54d79473a5a9eeb518752e33d0f1faee0f17b149a6ac205
|
3 |
+
size 14262
|
rng_state_11.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4c94abbc1d333765a5585d72a643f24cae81ca7fe1161a874e7dcaabbf7bd29d
|
3 |
+
size 14262
|
rng_state_12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8b56499538e7d62fbd4d2bff54cbb667287939a0f6a94bf9e1cb09c201c3768c
|
3 |
+
size 14262
|
rng_state_13.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:06acfc0f3525f99c12bb83e8c544b9326b328c5a4d2ef517cb977b9794bebe4c
|
3 |
+
size 14262
|
rng_state_14.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:437d4f5244d715af7f41df1f8dd9cd7c280e626d93971bf9615875624142ce4a
|
3 |
+
size 14262
|
rng_state_15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b7646ef496596683994b711ca3cc7e183550944e94fb5ca099b386f1b40861c5
|
3 |
+
size 14262
|
rng_state_16.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1c889ad312c69a6bc978e18ff28c3ee88a5123c118d1d54af319fed32de920f5
|
3 |
+
size 14262
|
rng_state_17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d0e6b4ebb5992685933e5e9eae6dc704358b5b430ee7474261ee21e4da4bf946
|
3 |
+
size 14262
|
rng_state_18.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ccd8bfc980810b4fd4cdab925c4cff5e8864e666db57a8c602e36d96ea2f9c9b
|
3 |
+
size 14262
|