| { |
| "name": "reflexdiffusion-il-1000k", |
| "save_dir": "/data/nuplan_opensourcetest/ReflexDiffusion", |
| "train_set": "/mnt/nuplan_data/nuplan_cache/exp", |
| "train_set_list": "/data/nuplan_opensourcetest/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": "/data/nuplan_opensourcetest/ReflexDiffusion/normalization.json", |
| "use_data_augment": true, |
| "num_workers": 4, |
| "pin_mem": true, |
| "seed": 3407, |
| "train_epochs": 500, |
| "save_utd": 20, |
| "batch_size": 512, |
| "learning_rate": 0.00025, |
| "warm_up_epoch": 5, |
| "encoder_drop_path_rate": 0.1, |
| "decoder_drop_path_rate": 0.1, |
| "alpha_planning_loss": 1.0, |
| "device": "cuda", |
| "use_ema": true, |
| "use_all_feature": 1.0, |
| "encoder_depth": 3, |
| "decoder_depth": 3, |
| "num_heads": 6, |
| "hidden_dim": 192, |
| "diffusion_model_type": "x_start", |
| "predicted_neighbor_num": 10, |
| "resume_model_path": null, |
| "use_wandb": false, |
| "notes": "", |
| "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 |
| ] |
| } |
| } |
| } |
