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