CokeAnd1ce's picture
Upload 25 files
c523607 verified
seed: 42
run_root_dir: runs
save_checkpoint: true
vla_id: baseline
vlm_path: /home/linyihan/linyh/VLM/Qwen3-VL-2B-Instruct
vlm_model_id: Qwen3
default_image_size: 224
codebook_size: 16
use_latent: false
pretrained_checkpoint: /home/linyihan/linyh/LatentVLA/runs/baseline+aloha+Qwen3+type-training+20251226-161628
from_pretrained: true
data_root_dir: /home/linyihan/linyh/datasets/robotwin_latent
data_mix:
- aloha_move_can_pot
- aloha_move_playingcard_away
- aloha_pick_dual_bottles
- aloha_place_container_plate
shuffle_buffer_size: 128
image_aug: true
window_size: 25
use_wrist_image: true
use_proprio: true
use_pro_version: true
num_images_in_input: 2
action_tokenizer_ckpt: /home/linyihan/linyh/laq/results_libero/tokenizer_step_50000.pt
type: training
epochs: 10
max_steps: 10000
global_batch_size: 128
per_device_batch_size: 32
learning_rate: 0.0002
weight_decay: 0.01
max_grad_norm: 1.0
lr_scheduler_type: constant
warmup_ratio: 0.03
save_step: 20000
token_loss_weight: 0.0
use_token: false
use_val: false
wandb_project: vla
use_wandb: true