{
  "step": 80000,
  "args": {
    "base_model": "/data/base_model",
    "config_model": "/data/config_model",
    "dataset": "/data/dataset",
    "vl_model_path": null,
    "canonical_vl_model_name": "nvidia/Cosmos-Reason2-2B",
    "label_dir": "adaptive_chunk_labels",
    "output_dir": "/data/output",
    "experiment_name": "task0088_adaptive_chunk32_from_shift5_80k_gpu0_20260703_054903",
    "max_steps": 80000,
    "global_batch_size": 8,
    "gradient_accumulation_steps": 1,
    "num_workers": 8,
    "actor_lr": 2e-05,
    "weight_decay": 1e-05,
    "warmup_ratio": 0.05,
    "action_horizon": 32,
    "approach_stride": 2,
    "manipulation_stride": 1,
    "approach_token_weight": 1.0,
    "manipulation_token_weight": 1.4,
    "approach_gripper_weight": 0.6,
    "manipulation_gripper_weight": 3.0,
    "arm_weight": 1.0,
    "gripper_indices": "7,15",
    "episode_sampling_rate": 0.1,
    "shard_size": 1024,
    "num_shards_per_epoch": 100000,
    "save_steps": 20000,
    "logging_steps": 10,
    "max_grad_norm": 1.0,
    "seed": 42,
    "bf16": true,
    "tf32": true,
    "local_files_only": true,
    "use_percentiles": false,
    "clip_outliers": true,
    "state_dropout_prob": null,
    "rtc_training_prefix_steps": 0,
    "save_optimizer_state": false,
    "save_final_model": true,
    "dry_run_data": false,
    "dry_run_forward": false
  }
}