forked from qidiso/mxnet-retrain
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
54 lines (54 loc) · 1.14 KB
/
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
{
"export": {
"use_latest": 1
},
"common": {
"num_threads": 8,
"gpus": 0
},
"test": {
"use_latest": 1,
"classification_report_output": 1,
"test_batch_size": 10,
"confusion_matrix_output": 1
},
"finetune": {
"max_random_shear_ratio": 0.01,
"disp_batches": 10,
"wd": 1e-05,
"models": [
"imagenet1k-inception-v3"
],
"batch_size": 10,
"lr_factor": 0.1,
"random_mirror": 1,
"mom": 0.9,
"num_epochs": 20,
"auto_test": 1,
"optimizers": [
"sgd"
],
"num_active_layers": 1,
"max_random_aspect_ratio": 0.05,
"max_random_rotate_angle": 30,
"train_accuracy_graph_output": 1,
"lr": 0.0001,
"random_crop": 1,
"lr_step_epochs": 10
},
"data": {
"quality": 100,
"shuffle": 1,
"center_crop": 0
},
"ensemble": {
"models": [
"20180130074818-imagenet1k-nin-nadam-0003",
"20180130075252-imagenet1k-squeezenet-nadam-0003",
"20180131105109-imagenet1k-caffenet-nadam-0003"
],
"confusion_matrix_output": 1,
"ensemble_batch_size": 20,
"classification_report_output": 1
}
}