| { |
| "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 |
| } |
| } |
