-
Notifications
You must be signed in to change notification settings - Fork 70
/
experiment-rs.sh
executable file
·65 lines (59 loc) · 1.88 KB
/
experiment-rs.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
63
64
65
#!/bin/bash
export PYTHONPATH=`pwd`
echo $PYTHONPATH
source $1
exp=$2
gpu=$3
ARGS=${@:4}
group_examples_by_query_flag=''
if [[ $group_examples_by_query = *"True"* ]]; then
group_examples_by_query_flag="--group_examples_by_query"
fi
relation_only_flag=''
if [[ $relation_only = *"True"* ]]; then
relation_only_flag="--relation_only"
fi
use_action_space_bucketing_flag=''
if [[ $use_action_space_bucketing = *"True"* ]]; then
use_action_space_bucketing_flag='--use_action_space_bucketing'
fi
cmd="python3 -m src.experiments \
--data_dir $data_dir \
$exp \
--model $model \
--bandwidth $bandwidth \
--entity_dim $entity_dim \
--relation_dim $relation_dim \
--history_dim $history_dim \
--history_num_layers $history_num_layers \
--num_rollouts $num_rollouts \
--num_rollout_steps $num_rollout_steps \
--bucket_interval $bucket_interval \
--num_epochs $num_epochs \
--num_wait_epochs $num_wait_epochs \
--num_peek_epochs $num_peek_epochs \
--batch_size $batch_size \
--train_batch_size $train_batch_size \
--dev_batch_size $dev_batch_size \
--margin $margin \
--learning_rate $learning_rate \
--baseline $baseline \
--grad_norm $grad_norm \
--emb_dropout_rate $emb_dropout_rate \
--ff_dropout_rate $ff_dropout_rate \
--action_dropout_rate $action_dropout_rate \
--action_dropout_anneal_interval $action_dropout_anneal_interval \
--reward_shaping_threshold $reward_shaping_threshold \
$relation_only_flag \
--beta $beta \
--beam_size $beam_size \
--num_paths_per_entity $num_paths_per_entity \
$group_examples_by_query_flag \
$use_action_space_bucketing_flag \
--distmult_state_dict_path $distmult_state_dict_path \
--complex_state_dict_path $complex_state_dict_path \
--conve_state_dict_path $conve_state_dict_path \
--gpu $gpu \
$ARGS"
echo "Executing $cmd"
$cmd