TangYishan commited on
Commit
e7511ea
·
verified ·
1 Parent(s): c301b62

Upload config.full.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.full.yaml +68 -0
config.full.yaml ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ run_id: starvla_qwenoft_g1_sonic_smoke
2
+ run_root_dir: ./results/Checkpoints
3
+ seed: 42
4
+ wandb_entity: your_wandb_entity
5
+ wandb_project: starVLA_unitree_g1_wholebody
6
+ is_debug: false
7
+ version_id: '0.21'
8
+ framework:
9
+ name: QwenOFT
10
+ qwenvl:
11
+ base_vlm: ./playground/Pretrained_models/Qwen3.5-0.8B
12
+ attn_implementation: sdpa
13
+ action_model:
14
+ action_model_type: MLP
15
+ action_dim: 78
16
+ action_hidden_dim: 1024
17
+ future_action_window_size: 7
18
+ past_action_window_size: 0
19
+ state_dim: 72
20
+ action_horizon: 8
21
+ datasets:
22
+ vla_data:
23
+ dataset_py: lerobot_datasets
24
+ include_state: true
25
+ data_root_dir: playground/Datasets/UnitreeG1_WholeBody/lerobot
26
+ data_mix: unitree_g1_test_sonic
27
+ action_type: abs_qpos
28
+ action_mode: abs
29
+ sequential_step_sampling: false
30
+ per_device_batch_size: 1
31
+ load_all_data_for_training: true
32
+ obs_image_size:
33
+ - 224
34
+ - 224
35
+ delete_pause_frame: false
36
+ video_backend: torchvision_av
37
+ trainer:
38
+ max_train_steps: 1000
39
+ num_warmup_steps: 50
40
+ save_interval: 200
41
+ eval_interval: 500
42
+ learning_rate:
43
+ base: 2.0e-05
44
+ qwen_vl_interface: 1.0e-05
45
+ action_model: 0.0001
46
+ lr_scheduler_type: cosine_with_min_lr
47
+ scheduler_specific_kwargs:
48
+ min_lr: 1.0e-06
49
+ freeze_modules: ''
50
+ loss_scale:
51
+ vla: 1.0
52
+ vlm: 0.1
53
+ max_grad_norm: 1.0
54
+ weight_decay: 0.0
55
+ logging_frequency: 10
56
+ gradient_clipping: 1.0
57
+ gradient_accumulation_steps: 4
58
+ gradient_checkpointing: true
59
+ optimizer:
60
+ name: AdamW
61
+ betas:
62
+ - 0.9
63
+ - 0.95
64
+ eps: 1.0e-08
65
+ weight_decay: 1.0e-08
66
+ save_format: pt
67
+ config_yaml: ./examples/realRobots/UnitreeG1_WholeBody/step2_training/train_files/starvla_qwenoft_g1_sonic.yaml
68
+ output_dir: ./results/Checkpoints/starvla_qwenoft_g1_sonic_smoke