File size: 1,020 Bytes
f6cc954 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
{
"device": "cuda",
"run_name": "Evo1_700M_metaworld_v1_self_dataset_stage2",
"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_metaworld.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_metaworld_v1_self_dataset_stage2",
"resume": true,
"resume_path": "/share/taolin/checkpoints/Evo1_700M_metaworld_v1_self_dataset_stage1/step_20000",
"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
} |