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