-
Notifications
You must be signed in to change notification settings - Fork 24
/
Copy pathadv_gen_replay.cfg
48 lines (42 loc) · 1014 Bytes
/
adv_gen_replay.cfg
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# save options
out: ./out/adv_gen_replay_out
# data options
data_dir: ./data/nuscenes
data_version: trainval
split: val
val_size: 400 # num scenes
seq_interval: 10 # 5 seconds between sequences.
shuffle: False
num_workers: 0
batch_size: 1 # sets the approximate num agents across all scenes being optimized
# traffic model weights
ckpt: ./model_ckpt/traffic_model.pth
# how to operate
feasibility_check_sep: True
planner: 'ego' # i.e. replay
viz: True
save: True # save the scenarios
# optimization options
num_iters: 300
lr: 0.05
# init optim
init_loss_motion_prior_ext: 0.1
init_loss_match_ext: 10.0
# adv optim
loss_coll_veh: 20.0
loss_coll_veh_plan: 20.0
loss_coll_env: 20.0
loss_init_z: 0.5
loss_init_z_atk: 0.05
loss_motion_prior: 1.0
loss_motion_prior_atk: 0.005
loss_motion_prior_ext: 0.0001
loss_match_ext: 10.0
loss_adv_crash: 2.0
# sol optim
sol_future_len: 16
sol_loss_motion_prior: 0.005
sol_loss_coll_veh: 10.0
sol_loss_coll_env: 10.0
sol_loss_motion_prior_ext: 0.001
sol_loss_match_ext: 10.0