-
Notifications
You must be signed in to change notification settings - Fork 19
Expand file tree
/
Copy pathrun_gtrs_dense_evaluation_syn.sh
More file actions
38 lines (31 loc) · 1.33 KB
/
run_gtrs_dense_evaluation_syn.sh
File metadata and controls
38 lines (31 loc) · 1.33 KB
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
split=navtest
agent=gtrs_dense_vov
metric_cache_path="${NAVSIM_EXP_ROOT}/cache/${split}_metric_cache"
#--------------------------
export SYN_IDX=0 # 0, 1, 2, 3, 4
export SYN_GT=pdm # pdm, recovery
syn_imi=true # true, false
backbone_type=resnet # vov, resnet
#--------------------------
postfix=v1.0-${SYN_IDX}
[ "$syn_imi" = "true" ] && tag="" || tag="_rewards_only"
dir=train_gtrs_dense_${backbone_type}_syn_react_${SYN_GT}_${postfix}${tag}
echo "Starting evaluation script..."
for epoch in $(seq 0 49); do
padded_epoch=$(printf "%02d" $epoch)
experiment_name="${dir}/test-${padded_epoch}ep-${split}-16384"
ckpt="'${NAVSIM_EXP_ROOT}/${dir}/epoch=${padded_epoch}.ckpt'"
export SUBSCORE_PATH=${NAVSIM_EXP_ROOT}/${experiment_name}/epoch${epoch}_${split}.pkl; # save path for the scores
python ${NAVSIM_DEVKIT_ROOT}/navsim/planning/script/run_pdm_score_one_stage_gpu.py \
agent=$agent \
+combined_inference=false \
dataloader.params.batch_size=16 \
agent.checkpoint_path=${ckpt} \
agent.config.vocab_path=${NAVSIM_DEVKIT_ROOT}/traj_final/16384.npy \
agent.config.backbone_type=${backbone_type} \
trainer.params.precision=32 \
experiment_name=${experiment_name} \
+cache_path=null \
metric_cache_path=${metric_cache_path} \
train_test_split=${split}
done