-
Notifications
You must be signed in to change notification settings - Fork 19
/
scripts_training.sh
28 lines (20 loc) · 1.13 KB
/
scripts_training.sh
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
#===============================================================================
# Training Scripts
#===============================================================================
# === KITTI Val 1 Split ====
source dependencies/cuda_8.0_env
# First train the warmup
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=kitti_3d_warmup
# Then train the model with uncertainty and GrooMeD-NMS
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=groumd_nms
# === KITTI Val 2 Split ====
source dependencies/cuda_8.0_env
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=kitti_3d_warmup_split2
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=groumd_nms_split2
# === KITTI Test (Full) Split ====
source dependencies/cuda_8.0_env
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=kitti_3d_warmup_full_train_2
CUDA_VISIBLE_DEVICES=0 python -u scripts/train_rpn_3d.py --config=groumd_nms_full_train_2
#===============================================================================
# Ablation Studies
#===============================================================================