AutoVLA / training_config.json
mjf-su's picture
Add files using upload-large-folder tool
1aebaf7 verified
Raw
History Blame Contribute Delete
1.81 kB
{
"checkpoint": "/workspace/fms4navigation/models/checkpoint-1832",
"jsonl": "/workspace/fms4navigation/datasets/AutoVLA/CounterfactualVLA-train.jsonl",
"data_root": "/workspace/fms4navigation/datasets/AutoVLA/camera/camera_front_wide_120fov",
"eval_jsonl": "/workspace/fms4navigation/datasets/AutoVLA/CounterfactualVLA-val.jsonl",
"mode": "adaptive",
"config_path": "/workspace/fms4navigation/configs/prompt/CounterfactualVLA.yaml",
"chat_template_path": null,
"output_dir": "/workspace/fms4navigation/results/AutoVLA-sft",
"num_epochs": 1,
"max_steps": -1,
"per_device_train_batch_size": 16,
"per_device_eval_batch_size": null,
"gradient_accumulation_steps": 3,
"learning_rate": 1e-06,
"warmup_ratio": 0.03,
"lr_scheduler_type": "constant",
"completion_only_loss": true,
"bf16": true,
"fp16": false,
"flash_attn": false,
"gradient_checkpointing": true,
"seed": 42,
"only_complete": true,
"ade_threshold": 9.1189,
"ade_field": null,
"eval_n": 500,
"reaction_seconds": 2.0,
"prediction_horizon": 6.0,
"past_trajectory_step_sec": 0.25,
"fut_trajectory_step_sec": 0.25,
"save_steps": 500,
"save_total_limit": 3,
"save_milestones": 0,
"resume_from_checkpoint": false,
"save_model": true,
"eval_steps": null,
"use_lora": false,
"lora_r": 16,
"lora_alpha": 32,
"lora_dropout": 0.05,
"wandb_project": "matt-router",
"run_name": "AutoVLA",
"login_to_wandb": true,
"logging_steps": 10,
"activation_eval_n": 128,
"activation_eval_steps": null,
"hf_repo": null,
"hf_token": null,
"hf_private": false,
"dataloader_num_workers": 4,
"n_hard_train": 99982,
"n_easy_train": 100000,
"hard_fraction_train": 0.4999549959496355,
"n_hard_val": 250,
"n_easy_val": 250,
"timestamp_utc": "2026-05-19T13:03:22.968442Z"
}