-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrun_tasks.sh
executable file
·59 lines (55 loc) · 1.27 KB
/
run_tasks.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
#!/bin/bash
#
# To install the latest version of parallel, run
# (wget -O - pi.dk/3 || curl pi.dk/3/ || fetch -o - http://pi.dk/3) | bash
GPU_IDS=( 0 1 )
NUM_SIMS=10
ENCODING_TYPE_OPTS="position_encoding"
SHARE_TYPE_OPTS="layerwise"
NONLIN_OPTS="relu"
DIM_MEMORY_OPTS="10 20 50"
DIM_EMB_OPTS="10 20 50"
NUM_CACHES_OPTS="1 5"
INIT_STDDEV_OPTS=0.1
LEARNING_RATE_OPTS="0.01"
MAX_GRAD_NORM_OPTS=40
NUM_HOPS_OPTS="5"
NOISE_OPTS="0 10"
SEARCH_PROB_OPTS="1.00"
EXIT_PROB_OPTS="0.67"
INFORM_PROB_OPTS="0.50"
TASK_ID_OPTS="21"
SPLIT="tom_new_tasks"
SHA=$(git log --pretty=format:'%h' -n 1)
DATE=`date +%Y-%m-%d`
parallel -j ${#GPU_IDS[@]} \
'export CUDA_VISIBLE_DEVICES=$(({%} - 1)) && \
python main.py -te -ne 100 \
-nl {1} \
-et {2} \
-st {3} \
-de {4} \
-dm {5} \
-nc {6} \
-is {7} \
-lr {8} \
-gn {9} \
-nh {10} \
-d data/{12}/world_large_nex_1000_{11} \
-o results/{12} \
-t {13} \
-te' \
::: $NONLIN_OPTS `# 1` \
::: $ENCODING_TYPE_OPTS `# 2` \
::: $SHARE_TYPE_OPTS `# 3` \
::: $DIM_EMB_OPTS `# 4` \
::: $DIM_MEMORY_OPTS `# 5` \
::: $NUM_CACHES_OPTS `# 6` \
::: $INIT_STDDEV_OPTS `# 7` \
::: $LEARNING_RATE_OPTS `# 8` \
::: $MAX_GRAD_NORM_OPTS `# 9` \
::: $NUM_HOPS_OPTS `# 10` \
::: $NOISE_OPTS `# 11` \
::: $SPLIT `# 12` \
::: ${TASK_ID_OPTS} `# 13` \
::: {1..$NUM_SIMS} `# 14`