-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathconfig.json
35 lines (35 loc) · 916 Bytes
/
config.json
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
{
"paths":{
"cnn_save_path" : "save_models/cnn_rnn_attn/cnn.model",
"rnn_save_path" : "save_models/cnn_rnn_attn/rnn.model",
"save_loss_path" : "save_models/cnn_rnn_attn/loss.csv",
"label_list_file" : "dataset/labels.txt",
"train_dir" : "dataset/train/images/",
"train_labels_file" : "dataset/train/labels.csv",
"val_dir" : "dataset/val/images/",
"val_labels_file" : "dataset/val/labels.csv",
"test_dir" : "dataset/test/images",
"submission_file" : "submission.csv"
},
"model":{
"embed_size" : 128,
"hidden_size" : 512,
"total_size" : 64
},
"training":{
"n_epoch_1" : 15,
"n_epoch_2" : 20,
"lr_1" : 1e-3,
"lr_2" : 1e-4,
"lr_1_decay" : 1.6,
"lr_2_decay" : 1.4,
"load_intial" : false,
"batch_size" : 32,
"n_workers" : 4,
"use_gpu" : true,
"cuda_device" : 0,
"n_batch_print" : 50,
"attention" : true,
"draw_image" : false
}
}