| { |
| "name": "diffusion-planner-training", |
| "save_dir": ".", |
| "train_set": "/home/Humble/Data/data/datasets/Diffusion-Planner", |
| "train_set_list": "/home/Humble/Humble/EndtoEnd/YSH-Diffusion-Planner/diffusion_planner_training.json", |
| "future_len": 80, |
| "time_len": 21, |
| "agent_state_dim": 11, |
| "agent_num": 32, |
| "static_objects_state_dim": 10, |
| "static_objects_num": 5, |
| "lane_len": 20, |
| "lane_state_dim": 12, |
| "lane_num": 70, |
| "route_len": 20, |
| "route_state_dim": 12, |
| "route_num": 25, |
| "augment_prob": 0.5, |
| "normalization_file_path": "normalization.json", |
| "use_data_augment": true, |
| "num_workers": 16, |
| "pin_mem": true, |
| "seed": 3407, |
| "train_epochs": 500, |
| "save_utd": 10, |
| "batch_size": 512, |
| "learning_rate": 0.0005, |
| "warm_up_epoch": 5, |
| "encoder_drop_path_rate": 0.1, |
| "decoder_drop_path_rate": 0.1, |
| "alpha_planning_loss": 1.0, |
| "gradient_accumulation_steps": 4, |
| "device": "cuda", |
| "use_ema": true, |
| "use_compile": false, |
| "encoder_depth": 3, |
| "decoder_depth": 3, |
| "num_heads": 6, |
| "hidden_dim": 192, |
| "diffusion_model_type": "x_start", |
| "predicted_neighbor_num": 10, |
| "resume_model_path": "/home/Humble/Humble/EndtoEnd/YSH-Diffusion-Planner/training_log/diffusion-planner-training/2026-02-04-00:16:42", |
| "use_wandb": false, |
| "notes": "", |
| "use_mlflow": true, |
| "mlflow_tracking_uri": "https://dagshub.com/YSH-research/YSH-Diffusion-Planner.mlflow", |
| "mlflow_experiment_name": "diffusion-planner", |
| "use_dvclive": true, |
| "ddp": true, |
| "port": "22323", |
| "state_normalizer": { |
| "mean": [ |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ], |
| [ |
| [ |
| 10, |
| 0, |
| 0, |
| 0 |
| ] |
| ] |
| ], |
| "std": [ |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ], |
| [ |
| [ |
| 20, |
| 20, |
| 1, |
| 1 |
| ] |
| ] |
| ] |
| }, |
| "observation_normalizer": { |
| "ego_current_state": { |
| "mean": [ |
| 10.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0 |
| ], |
| "std": [ |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0 |
| ] |
| }, |
| "neighbor_agents_past": { |
| "mean": [ |
| 10.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0 |
| ], |
| "std": [ |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 1.0 |
| ] |
| }, |
| "static_objects": { |
| "mean": [ |
| 10.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0 |
| ], |
| "std": [ |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 1.0, |
| 1.0 |
| ] |
| }, |
| "lanes": { |
| "mean": [ |
| 10.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0 |
| ], |
| "std": [ |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 1.0, |
| 1.0 |
| ] |
| }, |
| "lanes_speed_limit": { |
| "mean": [ |
| 0.0 |
| ], |
| "std": [ |
| 20.0 |
| ] |
| }, |
| "route_lanes": { |
| "mean": [ |
| 10.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0, |
| 0.0 |
| ], |
| "std": [ |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 20.0, |
| 1.0, |
| 1.0, |
| 1.0, |
| 1.0 |
| ] |
| }, |
| "route_lanes_speed_limit": { |
| "mean": [ |
| 0.0 |
| ], |
| "std": [ |
| 20.0 |
| ] |
| } |
| }, |
| "guidance_fn": null |
| } |
