File size: 2,873 Bytes
706b6ae 5e652cc 706b6ae 5e652cc 706b6ae 5e652cc 706b6ae 5e652cc 706b6ae 5e652cc 706b6ae 5e652cc 706b6ae 5e652cc 706b6ae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | {
"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
} |