args_config.json

{
  "vla_path": "/home/jisoo_kim/.cache/huggingface/hub/models--openvla--openvla-7b/snapshots/31f090d05236101ebfc381b61c674dd4746d4ce0",
  "dataset_type": "robocasa",
  "libero_data_dir": "/scratch2/jisoo6687/libero/libero_goal_no_noops_track",
  "libero_task_suite": "libero_goal",
  "robocasa_data_dir": "/weka/jisookim/dataset/robocasa/datasets/single_stage_regenerate",
  "robocasa_task_suite": "robocasa",
  "data_root_dir": "datasets/rlds",
  "dataset_name": "robocasa",
  "run_root_dir": "/weka/jisookim/experiment/openvla/robocasa/ckpt/robocasa_1024_input_head_onlytrack_mlp_8_stride1_hidden1024_blocks4_b16",
  "shuffle_buffer_size": 100000,
  "window_stride": 1,
  "action_chunk_size": null,
  "tracking_tracks_root": "/weka/jisookim/dataset/robocasa/datasets/pointrack",
  "tracking_tracks_filename": "vertex_tracks_face_uniform.npy",
  "precomputed_statistics_path": "/weka/jisookim/dataset/robocasa/datasets/single_stage_regenerate/robocasa_statistics.json",
  "use_l1_regression": true,
  "use_diffusion": false,
  "num_diffusion_steps_train": 50,
  "use_film": false,
  "num_images_in_input": 3,
  "use_pointcloud_input": false,
  "pointcloud_input_num_points": 1024,
  "pointcloud_input_dim": 3,
  "use_proprio": true,
  "use_tracking_head": true,
  "tracking_dim": 3,
  "tracking_num_points": 1024,
  "tracking_hidden_dim": 1024,
  "tracking_num_blocks": 4,
  "tracking_head_type": "with_point_input",
  "tracking_loss_weight": 1.0,
  "tracking_label_key": null,
  "tracking_use_point_features": true,
  "tracking_point_hidden_dim": 1024,
  "tracking_use_pointcloud_input": true,
  "save_tracking_viz": true,
  "tracking_viz_freq": 2000,
  "tracking_viz_max_points": 2000,
  "tracking_viz_dir": "/weka/jisookim/experiment/openvla/robocasa/ckpt/robocasa_1024_input_head_onlytrack_mlp_8_stride1_hidden1024_blocks4_b16/videos",
  "normalize_pointcloud": true,
  "normalize_tracking": true,
  "batch_size": 8,
  "learning_rate": 0.0005,
  "lr_warmup_steps": 0,
  "num_steps_before_decay": 100000,
  "grad_accumulation_steps": 1,
  "max_steps": 300005,
  "use_val_set": false,
  "val_freq": 10000,
  "val_time_limit": 180,
  "save_freq": 5000,
  "save_latest_checkpoint_only": false,
  "resume": false,
  "resume_step": null,
  "resume_checkpoint_dir": null,
  "base_model_path": null,
  "image_aug": true,
  "diffusion_sample_freq": 50,
  "use_lora": true,
  "lora_rank": 32,
  "lora_dropout": 0.0,
  "merge_lora_during_training": true,
  "use_wandb": false,
  "use_tensorboard": true,
  "tensorboard_log_dir": "/weka/jisookim/experiment/openvla/robocasa/runs/tensorboard_robocasa_1024_input_head_onlytrack_mlp_8_stride1_hidden1024_blocks4_b16",
  "wandb_entity": "your-wandb-entity",
  "wandb_project": "your-wandb-project",
  "run_id_note": "robocasa_single_stage_regenerate_3image--proprio_state_b16_aug",
  "run_id_override": null,
  "wandb_log_freq": 10
}