Hugging Face's logo

wangrongjing
/
Beyond_Imitation_Checkpoint

Model card Files
xet
 Community
Beyond_Imitation_Checkpoint / diffusion_Checkpoint2 /diffusion_model_args.json
wangrongjing's picture
wangrongjing
Upload 16 files
f3e36ba verified
Raw History Blame Contribute Delete
8.22 kB
{
"name": "diffusion-planner-training",
"save_dir": ".",
"train_set": "/mnt/nuplan_data/nuplan_cache/exp",
"train_set_list": "/data/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": 4,
"pin_mem": true,
"seed": 3407,
"train_epochs": 500,
"save_utd": 20,
"batch_size": 2048,
"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,
"device": "cuda",
"use_ema": true,
"encoder_depth": 3,
"decoder_depth": 3,
"num_heads": 6,
"hidden_dim": 192,
"diffusion_model_type": "x_start",
"use_bidirectional": false,
"use_safety_aware": true,
"no_sad": false,
"no_rii": false,
"predicted_neighbor_num": 10,
"resume_model_path": "/data/Diffusion-Planner/training_log/diffusion-planner-training/2026-05-18-14:44:02",
"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
]
}
},
"use_sad": true,
"use_rii": true
}