-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.toml
49 lines (45 loc) · 1.3 KB
/
config.toml
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
# All major configurations for this project
[runtime]
# Select
# mode = "FER-train" for training
# = "FER-predict" and show emoji
# = "mt-train" for english to kannada translation model
# = "q2" for plots and AUC in q2
# = "mt-q2" for data explore of machine tranlation dataset in q2
# = "plot_model_history" to plot the model history from logs
mode = "FER-predict"
[dir]
figures = "figs"
data = "data"
output = "outputs"
FER-model = "fer-detector-model"
[img_info]
size = [48, 48]
num_classes = 7
[cnn]
epoch = 60
batch_size = 32
learning_rate = 0.0001
loss = "categorical_crossentropy"
metrics = ["accuracy"]
# checkpoint file name
chk_path = "cp.ckpt"
csv_log = "training.log"
# Settings for machine translation task
[mt]
dataset = "en-kn-large"
max_sen_word_limit = 5
max_sen_token_size = 7
# Select
# model_name = "simple-rnn-model" for training on simple rnn
# = "gru-model" for training on gru
# = "gru-wo-embed-model" for training on gru
# = "lstm-model" for training on gru
model_name = "lstm-model"
epochs = 150
batch_size = 64
val_split = 0.2
learning_rate = 0.001
embedding_size = 128
num_rnn_units = 64
dropout_rate = 0.2