-
Notifications
You must be signed in to change notification settings - Fork 34
/
config.sh
43 lines (43 loc) · 1.05 KB
/
config.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
#!/bin/bash
host=`hostname`
experiment_dir="../"
output_dir="results"
experiment_file=$experiment_dir/0.txt
output_dir=$output_dir/lse
data_dir="../examples/data_small_output/_architecture_structure_address/"
gpu_id=-1
numEntityTypes=7
includeEntityTypes=1
includeEntity=0
numEpoch=20
numFeatureTemplates=10
rnnHidSize=250
relationEmbeddingDim=250
entityTypeEmbeddingDim=100
entityEmbeddingDim=50
relationVocabSize=51390
entityVocabSize=1542690
entityTypeVocabSize=2218
topK=2 #0 is max; 1 is top K , 2 is LogSumExp
K=5
regularize=0
learningRate=1e-3
learningRateDecay=0.0167 #(1/60)
l2=1e-3
rnnType='rnn' #rnn or lstm as of now
epsilon=1e-8 #epsilon for adam
gradClipNorm=5
gradientStepCounter=100000 #to print loss after gradient updates
saveFrequency=1
batchSize=32
useGradClip=1 # 0 == L2 regularization
# package_path='/home/rajarshi/ChainsofReasoning/model/?.lua'
useAdam=1
paramInit=0.1
evaluationFrequency=5
createExptDir=1 #make it 0 if you dont want to create a directory and only print stuff
useReLU=1
rnnInitialization=1
numLayers=1
useDropout=0
dropout=0.3