|
{ |
|
"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" |
|
} |
|
} |
|
} |
|
|