| {"DVI":{ | |
| "SETTING": "abnormal", | |
| "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_with_dropout", | |
| "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" | |
| } | |
| }} |