File size: 2,804 Bytes
d5fa9ad
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
{
    "SETTING": "normal",
    "CLASSES": ["plane", "car", "bird", "cat", "deer", "dog", "frog", "horse", "ship", "truck"], 
    "DATASET": "cifar10",
    "EPOCH_START": 1,
    "EPOCH_END": 200,
    "EPOCH_PERIOD": 1,
    "GPU":0,
    "TRAINING": {
        "NET": "resnet18",
        "loader_tr_args": {"batch_size": 128, "num_workers": 1},
        "loader_te_args": {"batch_size": 1000, "num_workers": 1},
        "optimizer_args": {"lr": 0.1, "momentum": 0.9, "weight_decay": 5e-4},
        "num_class": 10,
        "train_num": 50000,
        "test_num": 10000,
        "milestone":[160]
    },
    "VISUALIZATION":{

        "S_LAMBDA":1,
        "PREPROCESS":0,
        "BOUNDARY":{
            "B_N_EPOCHS": 0,
            "L_BOUND":0.6
        },
        "INIT_NUM":300,

        "ALPHA":0,
        "BETA":0.1,
        "MAX_HAUSDORFF":0.4,

        "LAMBDA": 10.0,
        "HIDDEN_LAYER":4,
        "ENCODER_DIMS": [512,256,256,256,256,2],
        "DECODER_DIMS": [2,256,256,256,256,512],
        "N_NEIGHBORS":15,
        "MAX_EPOCH": 20,
        "S_N_EPOCHS": 5,
        "T_N_EPOCHS": 100,
        "PATIENT": 3,
        "RESOLUTION":300,
        "VIS_MODEL_NAME": "vis",
        "EVALUATION_NAME": "evalution"
    },
    "DVI": {
        "SETTING": "normal",
        "CLASSES": ["0"], 
        "DATASET": "cifar10",
        "GPU": "0",
        "EPOCH_START": 1,
        "EPOCH_END": 2,
        "EPOCH_PERIOD": 1,
        "EPOCH_NAME":"Epoch",
        "TRAINING": {
            "NET": "resnet18",
            "loader_tr_args": {
                "batch_size": 128,
                "num_workers": 1
            },
            "loader_te_args": {
                "batch_size": 1000,
                "num_workers": 1
            },
            "optimizer_args": {
                "lr": 0.1,
                "momentum": 0.9,
                "weight_decay": 0.0005
            },
            "num_class": 1,
            "train_num": 24927,
            "test_num": 1261,
            "milestone": [
                10
            ]
        },
        "VISUALIZATION": {
            "PREPROCESS": 1,
            "BOUNDARY": {
                "B_N_EPOCHS": 0,
                "L_BOUND": 0.4
            },
            "BATCH_SIZE":1000,
            "LAMBDA1": 5.0,
            "LAMBDA2": 0.3,
       
            "ENCODER_DIMS_O": [128,64,64,64,64,2],
            "DECODER_DIMS_O": [2,64,64,64,64,128],
            "ENCODER_DIMS": [768,256,256,256,256,2],
            "DECODER_DIMS": [2,256,256,256,256,768],
            "N_NEIGHBORS": 15,
            "MAX_EPOCH": 20,
            "S_N_EPOCHS": 10,
            "PATIENT": 3,
            "RESOLUTION": 300,
            "VIS_MODEL_NAME": "dvi",
            "FLAG": "_temporal_id_withoutB",
            "EVALUATION_NAME": "evaluation_tfDVI"
        }
    }
}