| { | |
| "device": "cuda", | |
| "run_name": "Evo1_700M_libero_4suite_stage2_v2", | |
| "vlm_name": "OpenGVLab/InternVL3-1B", | |
| "action_head": "flowmatching", | |
| "return_cls_only": false, | |
| "disable_wandb": true, | |
| "dataset_type": "lerobot", | |
| "data_paths": null, | |
| "dataset_config_path": "dataset/config_libero.yaml", | |
| "image_size": 448, | |
| "binarize_gripper": false, | |
| "use_augmentation": true, | |
| "lr": 1e-05, | |
| "batch_size": 16, | |
| "max_steps": 80000, | |
| "warmup_steps": 1000, | |
| "grad_clip_norm": 1.0, | |
| "weight_decay": 0.001, | |
| "log_interval": 10, | |
| "ckpt_interval": 2500, | |
| "save_dir": "/share/taolin/checkpoints/Evo1_700M_libero_4suite_stage2_v2", | |
| "resume": true, | |
| "resume_path": "/share/taolin/checkpoints/Evo1_700M_libero_4suite_stage1_v2/step_10000", | |
| "resume_pretrain": true, | |
| "use_vggt": false, | |
| "finetune_vlm": true, | |
| "finetune_action_head": true, | |
| "per_action_dim": 24, | |
| "state_dim": 24, | |
| "horizon": 50, | |
| "num_layers": 8, | |
| "num_workers": 4, | |
| "dropout": 0.2, | |
| "action_dim": 1200 | |
| } |