saved_models/codesearch_contrastive_learning/config_dvi_modi.json

{
    "SETTING": "normal",
    "CLASSES": [
        "0"
    ],
    "GPU": "0",
    "DATASET": "cifar10",
    "EPOCH_START": 1,
    "EPOCH_END": 1,
    "EPOCH_PERIOD": 1,
    "TRAINING": {
        "NET": "resnet18",
        "num_class": 1,
        "train_num": 24927,
        "test_num": 1261
    },
    "VISUALIZATION": {
        "PREPROCESS": 1,
        "BOUNDARY": {
            "B_N_EPOCHS": 0,
            "L_BOUND": 0.5
        },
        "INIT_NUM": 300,
        "ALPHA": 1,
        "BETA": 1,
        "MAX_HAUSDORFF": 0.33,
        "LAMBDA1": 1,
        "LAMBDA2": 0.3,
        "S_LAMBDA": 1,
        "ENCODER_DIMS": [
            768,
            256,
            256,
            256,
            256,
            2
        ],
        "DECODER_DIMS": [
            2,
            256,
            256,
            256,
            256,
            768
        ],
        "N_NEIGHBORS": 15,
        "MAX_EPOCH": 200,
        "S_N_EPOCHS": 5,
        "T_N_EPOCHS": 20,
        "PATIENT": 3,
        "RESOLUTION": 300,
        "VIS_MODEL_NAME": "dvi",
        "EVALUATION_NAME": "test_evaluation_dvi"
    }
}