lihao3
Add model checkpoint
42ecab9
action_dim: 7
action_model_type: DiT-B
data_root_dir: ./LIBERO/dataset
debug: false
future_action_window_size: 7
hf_token: hf_token
image_aug: true
is_resume: false
load_all_data_for_training: true
past_action_window_size: 3
pretrained_checkpoint: path/to/ckpt
repeated_diffusion_steps: 4
resume_epoch: null
resume_step: null
run_id: HIS_prism-dinosiglip-224px+LIBERO_goal+diffusion+n8+b16+x42--image_aug
run_id_note: null
run_root_dir: outputs/HIS_ditb_8_lr2e-5_b16_fa15_pa0_shuffle_LIBERO_goal
save_interval: 1500
seed: 42
trackers:
- jsonl
- wandb
use_ema: false
vla:
base_vlm: prism-dinosiglip-224px+7b
data_mix: libero_goal_no_noops
enable_gradient_checkpointing: true
enable_mixed_precision_training: true
epochs: 100
expected_world_size: 16
freeze_llm_backbone: false
freeze_vision_backbone: false
global_batch_size: 256
learning_rate: 2.0e-05
lr_scheduler_type: constant
max_grad_norm: 1.0
max_steps: 40000
per_device_batch_size: 16
reduce_in_full_precision: true
shuffle_buffer_size: 250000
train_strategy: fsdp-full-shard
type: prism-dinosiglip-224px+oxe+diffusion
unfreeze_last_llm_layer: false
vla_id: prism-dinosiglip-224px+oxe+diffusion
warmup_ratio: 0.0
weight_decay: 0.0
wandb_entity:
wandb_project: