-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathlaunch_image_data.sh
executable file
·62 lines (49 loc) · 1.81 KB
/
launch_image_data.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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#!/bin/bash
export SYSTEM='carla'
export ROOT='./data/carla/images/Dataset_with_image/left_turn_state_space_sampling'
export FNAME='Data_Collection_Compiled.pd'
# middle of turn
export DATA_PATH_1="${ROOT}/random_noise_driving/${FNAME}"
export DATA_PATH_2="${ROOT}/straight_lane_driving/${FNAME}"
export DATA_PATH_3="${ROOT}/start_of_turn/${FNAME}"
export DATA_PATH_4="${ROOT}/middle_of_turn/${FNAME}"
export RESULTS_PATH='./results-tmp'
# Margins for optimization constraints
export GAMMA_SAFE=0.05
export GAMMA_UNSAFE=0.05
export GAMMA_DYN=0.01
# Lagrange multipliers (fixed)
export LAMBDA_GRAD=0.01
export LAMBDA_PARAM=0.01
# Robustness
export DELTA_F=0.1
export DELTA_G=0.1
# Output map
# Choose the best LIP_CONST_1 & 2 based on your case
export LIP_CONST_1=0.1
export LIP_CONST_2=0.1
# Training
export NET_DIMS=(32 16)
export N_EPOCHS=2000
export LEARNING_RATE=0.005
export DUAL_STEP_SIZE=0.05
export DUAL_SCHEME='ae'
# For clean data with data aug
export NEIGHBOR_THRESH=0.008
export MIN_N_NEIGHBORS=200
# Additional state sampling
export N_SAMP_UNSAFE=0
export N_SAMP_SAFE=0
export N_SAMP_ALL=0
python main.py \
--system $SYSTEM --data-path $DATA_PATH_1 $DATA_PATH_2 $DATA_PATH_3 $DATA_PATH_4 \
--results-path $RESULTS_PATH \
--gamma-safe $GAMMA_SAFE --gamma-unsafe $GAMMA_UNSAFE --gamma-dyn $GAMMA_DYN \
--lambda-grad $LAMBDA_GRAD --lambda-param $LAMBDA_PARAM \
--net-dims ${NET_DIMS[@]} --n-epochs $N_EPOCHS \
--learning-rate $LEARNING_RATE --dual-step-size $DUAL_STEP_SIZE \
--nbr-thresh $NEIGHBOR_THRESH --min-n-nbrs $MIN_N_NEIGHBORS \
--n-samp-unsafe $N_SAMP_UNSAFE --n-samp-safe $N_SAMP_SAFE --n-samp-all $N_SAMP_ALL \
--dual-scheme $DUAL_SCHEME \
--robust --delta-f $DELTA_F --delta-g $DELTA_G \
--use-lip-output-term --lip-const-a $LIP_CONST_1 --lip-const-b $LIP_CONST_2