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