{
  "action_dim": 7,
  "action_tokenizer_exist": false,
  "class_dropout_prob": 0.0,
  "data_root_dir": "/media/liuzhuoyang/data/rlbench/rlds/12tasks_selected_keyframe_pointcloud_0814",
  "future_action_window_size": 0,
  "hf_token": ".hf_token",
  "image_aug": false,
  "is_resume": false,
  "llm_vision_layers": 8,
  "load_all_data_for_training": true,
  "past_action_window_size": 0,
  "pretrained_checkpoint": "/media/liuzhuoyang/new_vla/Rec_Diff_beta/exp/exp_12tasks_selected_keyframe_pointcloud_0814_Pretrainrtx0818e1_FreezeVistrue_Window0_Difftrue_Recfalse_Contrastive_Vislayer8_1024_0403_0820/checkpoints/step-005415-epoch-300-loss=1.4414.pt",
  "recon_image": true,
  "recon_pointcloud": true,
  "repeated_diffusion_steps": 4,
  "resume_epoch": null,
  "resume_step": null,
  "run_id": "exp_12tasks_selected_keyframe_pointcloud_0814_Pretraindiff300_FreezeVistrue_Window0_Difftrue_PCtrue_Rectrue_Contrastive_Vislayer8_1024_0403_0822",
  "run_id_note": null,
  "run_root_dir": "/media/liuzhuoyang/new_vla/Rec_Diff_beta/exp",
  "save_interval": 100,
  "seed": 42,
  "trackers": [
    "jsonl",
    "wandb"
  ],
  "use_contrastive": true,
  "use_diff": true,
  "use_ema": false,
  "use_pointcloud": true,
  "use_reconstruction": true,
  "use_roi": false,
  "vla": {
    "base_vlm": "prism-dinosiglip-224px+7b",
    "data_mix": "rlbench",
    "enable_gradient_checkpointing": true,
    "enable_mixed_precision_training": true,
    "epochs": 300,
    "expected_world_size": 32,
    "freeze_llm_backbone": false,
    "freeze_vision_tower": true,
    "global_batch_size": 256,
    "learning_rate": 2e-05,
    "lr_scheduler_type": "constant",
    "max_grad_norm": 1.0,
    "max_steps": null,
    "per_device_batch_size": 8,
    "reduce_in_full_precision": true,
    "shuffle_buffer_size": 10000,
    "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": "liumail2023-peking-university",
  "wandb_project": "one_model_vla_sft"
}