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