{
    "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,
    "map_len": 10,
    "map_state_dim": 4,
    "map_num": 5,
    "route_len": 20,
    "route_state_dim": 12,
    "route_num": 25,
    "encoder_drop_path_rate": 0.1,
    "decoder_drop_path_rate": 0.1,
    "device": "cuda",
    "encoder_depth": 3,
    "decoder_depth": 3,
    "num_heads": 6,
    "hidden_dim": 192,
    "diffusion_model_type": "x_start",
    "predicted_neighbor_num": 10,
    "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
            ]
        }
    }
}