Upload 8 files
Browse files- Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/checkpoint.json +5 -0
- Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/config.json +36 -0
- Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/mp_rank_00_model_states.pt +3 -0
- Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/norm_stats.json +46 -0
- Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/checkpoint.json +5 -0
- Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/config.json +36 -0
- Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/mp_rank_00_model_states.pt +3 -0
- Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/norm_stats.json +32 -0
Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/checkpoint.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"type": "ds_model",
|
| 3 |
+
"version": 0.0,
|
| 4 |
+
"checkpoints": "mp_rank_00_model_states.pt"
|
| 5 |
+
}
|
Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"device": "cuda",
|
| 3 |
+
"run_name": "Evo1_700M_libero_4suite_stage2_v2",
|
| 4 |
+
"vlm_name": "OpenGVLab/InternVL3-1B",
|
| 5 |
+
"action_head": "flowmatching",
|
| 6 |
+
"return_cls_only": false,
|
| 7 |
+
"disable_wandb": true,
|
| 8 |
+
"dataset_type": "lerobot",
|
| 9 |
+
"data_paths": null,
|
| 10 |
+
"dataset_config_path": "dataset/config_libero.yaml",
|
| 11 |
+
"image_size": 448,
|
| 12 |
+
"binarize_gripper": false,
|
| 13 |
+
"use_augmentation": true,
|
| 14 |
+
"lr": 1e-05,
|
| 15 |
+
"batch_size": 16,
|
| 16 |
+
"max_steps": 80000,
|
| 17 |
+
"warmup_steps": 1000,
|
| 18 |
+
"grad_clip_norm": 1.0,
|
| 19 |
+
"weight_decay": 0.001,
|
| 20 |
+
"log_interval": 10,
|
| 21 |
+
"ckpt_interval": 2500,
|
| 22 |
+
"save_dir": "/share/taolin/checkpoints/Evo1_700M_libero_4suite_stage2_v2",
|
| 23 |
+
"resume": true,
|
| 24 |
+
"resume_path": "/share/taolin/checkpoints/Evo1_700M_libero_4suite_stage1_v2/step_10000",
|
| 25 |
+
"resume_pretrain": true,
|
| 26 |
+
"use_vggt": false,
|
| 27 |
+
"finetune_vlm": true,
|
| 28 |
+
"finetune_action_head": true,
|
| 29 |
+
"per_action_dim": 24,
|
| 30 |
+
"state_dim": 24,
|
| 31 |
+
"horizon": 50,
|
| 32 |
+
"num_layers": 8,
|
| 33 |
+
"num_workers": 4,
|
| 34 |
+
"dropout": 0.2,
|
| 35 |
+
"action_dim": 1200
|
| 36 |
+
}
|
Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/mp_rank_00_model_states.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1bc3abd53046648ef72970f338753fadf5acf3f444117a7e3b36e1ff93f71ef0
|
| 3 |
+
size 1552487120
|
Evo1_Simulation_Benchmark_Checkpoints/LIBERO/Evo1_LIBERO_checkpoint/norm_stats.json
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"libero_robot": {
|
| 3 |
+
"observation.state": {
|
| 4 |
+
"min": [
|
| 5 |
+
-0.4828203022480011,
|
| 6 |
+
-0.3255046010017395,
|
| 7 |
+
0.008128180168569088,
|
| 8 |
+
0.35277295112609863,
|
| 9 |
+
-3.641430377960205,
|
| 10 |
+
-1.842738389968872,
|
| 11 |
+
-0.0013586411951109767,
|
| 12 |
+
-0.042040832340717316
|
| 13 |
+
],
|
| 14 |
+
"max": [
|
| 15 |
+
0.21031762659549713,
|
| 16 |
+
0.39128610491752625,
|
| 17 |
+
1.3660105466842651,
|
| 18 |
+
3.6714255809783936,
|
| 19 |
+
3.560650587081909,
|
| 20 |
+
1.386339545249939,
|
| 21 |
+
0.04233968257904053,
|
| 22 |
+
0.0013633022317662835
|
| 23 |
+
]
|
| 24 |
+
},
|
| 25 |
+
"action": {
|
| 26 |
+
"min": [
|
| 27 |
+
-0.9375,
|
| 28 |
+
-0.9375,
|
| 29 |
+
-0.9375,
|
| 30 |
+
-0.2582142949104309,
|
| 31 |
+
-0.375,
|
| 32 |
+
-0.3675000071525574,
|
| 33 |
+
0.0
|
| 34 |
+
],
|
| 35 |
+
"max": [
|
| 36 |
+
0.9375,
|
| 37 |
+
0.9375,
|
| 38 |
+
0.9375,
|
| 39 |
+
0.3557142913341522,
|
| 40 |
+
0.375,
|
| 41 |
+
0.375,
|
| 42 |
+
1.0
|
| 43 |
+
]
|
| 44 |
+
}
|
| 45 |
+
}
|
| 46 |
+
}
|
Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/checkpoint.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"type": "ds_model",
|
| 3 |
+
"version": 0.0,
|
| 4 |
+
"checkpoints": "mp_rank_00_model_states.pt"
|
| 5 |
+
}
|
Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"device": "cuda",
|
| 3 |
+
"run_name": "Evo1_700M_metaworld_v1_self_dataset_stage2",
|
| 4 |
+
"vlm_name": "OpenGVLab/InternVL3-1B",
|
| 5 |
+
"action_head": "flowmatching",
|
| 6 |
+
"return_cls_only": false,
|
| 7 |
+
"disable_wandb": true,
|
| 8 |
+
"dataset_type": "lerobot",
|
| 9 |
+
"data_paths": null,
|
| 10 |
+
"dataset_config_path": "dataset/config_metaworld.yaml",
|
| 11 |
+
"image_size": 448,
|
| 12 |
+
"binarize_gripper": false,
|
| 13 |
+
"use_augmentation": true,
|
| 14 |
+
"lr": 1e-05,
|
| 15 |
+
"batch_size": 16,
|
| 16 |
+
"max_steps": 80000,
|
| 17 |
+
"warmup_steps": 1000,
|
| 18 |
+
"grad_clip_norm": 1.0,
|
| 19 |
+
"weight_decay": 0.001,
|
| 20 |
+
"log_interval": 10,
|
| 21 |
+
"ckpt_interval": 2500,
|
| 22 |
+
"save_dir": "/share/taolin/checkpoints/Evo1_700M_metaworld_v1_self_dataset_stage2",
|
| 23 |
+
"resume": true,
|
| 24 |
+
"resume_path": "/share/taolin/checkpoints/Evo1_700M_metaworld_v1_self_dataset_stage1/step_20000",
|
| 25 |
+
"resume_pretrain": true,
|
| 26 |
+
"use_vggt": false,
|
| 27 |
+
"finetune_vlm": true,
|
| 28 |
+
"finetune_action_head": true,
|
| 29 |
+
"per_action_dim": 24,
|
| 30 |
+
"state_dim": 24,
|
| 31 |
+
"horizon": 50,
|
| 32 |
+
"num_layers": 8,
|
| 33 |
+
"num_workers": 4,
|
| 34 |
+
"dropout": 0.2,
|
| 35 |
+
"action_dim": 1200
|
| 36 |
+
}
|
Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/mp_rank_00_model_states.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:38ce9c71ab10ea6ac1e51e451ad277288f269eb341a8e559c20c4f29f999c4d3
|
| 3 |
+
size 1552487184
|
Evo1_Simulation_Benchmark_Checkpoints/MetaWorld/Evo1_MetaWorld_checkpoint/norm_stats.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metaworld_robot": {
|
| 3 |
+
"observation.state": {
|
| 4 |
+
"min": [
|
| 5 |
+
-0.498945415019989,
|
| 6 |
+
0.38712382316589355,
|
| 7 |
+
0.04404452443122864,
|
| 8 |
+
0.2622213363647461
|
| 9 |
+
],
|
| 10 |
+
"max": [
|
| 11 |
+
0.47120678424835205,
|
| 12 |
+
0.89215487241745,
|
| 13 |
+
0.4682557284832001,
|
| 14 |
+
1.0
|
| 15 |
+
]
|
| 16 |
+
},
|
| 17 |
+
"action": {
|
| 18 |
+
"min": [
|
| 19 |
+
-10.850606918334961,
|
| 20 |
+
-15.16084098815918,
|
| 21 |
+
-17.020706176757812,
|
| 22 |
+
-1.0
|
| 23 |
+
],
|
| 24 |
+
"max": [
|
| 25 |
+
19.347152709960938,
|
| 26 |
+
12.40846061706543,
|
| 27 |
+
24.61577606201172,
|
| 28 |
+
1.0
|
| 29 |
+
]
|
| 30 |
+
}
|
| 31 |
+
}
|
| 32 |
+
}
|