File size: 1,789 Bytes
82601b0 |
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 |
{
"DVI": {
"SETTING": "normal",
"CLASSES": [
"0",
"1",
"2",
"3",
"4",
"5",
"6",
"7",
"8",
"9"
],
"GPU": "0",
"DATASET": "mnist",
"EPOCH_START": 1,
"EPOCH_END": 20,
"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": 10,
"train_num": 60000,
"test_num": 10000,
"milestone": [
10
]
},
"VISUALIZATION": {
"PREPROCESS": 0,
"BOUNDARY": {
"B_N_EPOCHS": 1,
"L_BOUND": 0.4
},
"BATCH_SIZE":1000,
"LAMBDA1": 1.0,
"LAMBDA2": 0.3,
"ENCODER_DIMS": [
512,
256,
256,
256,
2
],
"DECODER_DIMS": [
2,
256,
256,
256,
512
],
"N_NEIGHBORS": 15,
"MAX_EPOCH": 20,
"S_N_EPOCHS": 10,
"PATIENT": 3,
"RESOLUTION": 300,
"FLAG": "_temporal_id_withoutB",
"EVALUATION_NAME": "proxy_eval"
}
}
} |