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