config.json

{"VERSION": 1, "OUTPUT_DIR": "output/imagenet_ood/MaPLe_OOD/vit_b16_c2_ep5_batch4_2ctx_cross_datasets_16shots/seed1", "RESUME": "", "SEED": 1, "USE_CUDA": true, "VERBOSE": true, "INPUT": {"SIZE": [224, 224], "INTERPOLATION": "bicubic", "TRANSFORMS": ["random_resized_crop", "random_flip", "normalize"], "NO_TRANSFORM": false, "PIXEL_MEAN": [0.48145466, 0.4578275, 0.40821073], "PIXEL_STD": [0.26862954, 0.26130258, 0.27577711], "CROP_PADDING": 4, "RRCROP_SCALE": [0.08, 1.0], "CUTOUT_N": 1, "CUTOUT_LEN": 16, "GN_MEAN": 0.0, "GN_STD": 0.15, "RANDAUGMENT_N": 2, "RANDAUGMENT_M": 10, "COLORJITTER_B": 0.4, "COLORJITTER_C": 0.4, "COLORJITTER_S": 0.4, "COLORJITTER_H": 0.1, "RGS_P": 0.2, "GB_P": 0.5, "GB_K": 21}, "DATASET": {"ROOT": "data/", "NAME": "ImageNet_OOD", "SOURCE_DOMAINS": [], "TARGET_DOMAINS": [], "NUM_LABELED": -1, "NUM_SHOTS": 16, "VAL_PERCENT": 0.1, "STL10_FOLD": -1, "CIFAR_C_TYPE": "", "CIFAR_C_LEVEL": 1, "ALL_AS_UNLABELED": false, "SUBSAMPLE_CLASSES": "all"}, "DATALOADER": {"NUM_WORKERS": 8, "K_TRANSFORMS": 1, "RETURN_IMG0": false, "TRAIN_X": {"SAMPLER": "RandomSampler", "BATCH_SIZE": 128, "N_DOMAIN": 0, "N_INS": 16}, "TRAIN_U": {"SAME_AS_X": true, "SAMPLER": "RandomSampler", "BATCH_SIZE": 32, "N_DOMAIN": 0, "N_INS": 16}, "TEST": {"SAMPLER": "SequentialSampler", "BATCH_SIZE": 100}}, "MODEL": {"INIT_WEIGHTS": "", "BACKBONE": {"NAME": "ViT-B/16", "PRETRAINED": true}, "HEAD": {"NAME": "", "HIDDEN_LAYERS": [], "ACTIVATION": "relu", "BN": true, "DROPOUT": 0.0}}, "OPTIM": {"NAME": "sgd", "LR": 0.01, "WEIGHT_DECAY": 0.0005, "MOMENTUM": 0.9, "SGD_DAMPNING": 0, "SGD_NESTEROV": false, "RMSPROP_ALPHA": 0.99, "ADAM_BETA1": 0.9, "ADAM_BETA2": 0.999, "STAGED_LR": false, "NEW_LAYERS": [], "BASE_LR_MULT": 0.1, "LR_SCHEDULER": "cosine", "STEPSIZE": [-1], "GAMMA": 0.1, "MAX_EPOCH": 2, "WARMUP_EPOCH": 1, "WARMUP_TYPE": "constant", "WARMUP_CONS_LR": 1e-05, "WARMUP_MIN_LR": 1e-05, "WARMUP_RECOUNT": true}, "TRAIN": {"CHECKPOINT_FREQ": 0, "PRINT_FREQ": 20, "COUNT_ITER": "train_x"}, "TEST": {"EVALUATOR": "Classification", "PER_CLASS_RESULT": false, "COMPUTE_CMAT": false, "NO_TEST": false, "SPLIT": "test", "FINAL_MODEL": "last_step"}, "TRAINER": {"NAME": "MaPLe_OOD", "MCD": {"N_STEP_F": 4}, "MME": {"LMDA": 0.1}, "CDAC": {"CLASS_LR_MULTI": 10, "RAMPUP_COEF": 30, "RAMPUP_ITRS": 1000, "TOPK_MATCH": 5, "P_THRESH": 0.95, "STRONG_TRANSFORMS": []}, "SE": {"EMA_ALPHA": 0.999, "CONF_THRE": 0.95, "RAMPUP": 300}, "M3SDA": {"LMDA": 0.5, "N_STEP_F": 4}, "DAEL": {"WEIGHT_U": 0.5, "CONF_THRE": 0.95, "STRONG_TRANSFORMS": []}, "CROSSGRAD": {"EPS_F": 1.0, "EPS_D": 1.0, "ALPHA_F": 0.5, "ALPHA_D": 0.5}, "DDAIG": {"G_ARCH": "", "LMDA": 0.3, "CLAMP": false, "CLAMP_MIN": -1.0, "CLAMP_MAX": 1.0, "WARMUP": 0, "ALPHA": 0.5}, "DAELDG": {"WEIGHT_U": 0.5, "CONF_THRE": 0.95, "STRONG_TRANSFORMS": []}, "DOMAINMIX": {"TYPE": "crossdomain", "ALPHA": 1.0, "BETA": 1.0}, "ENTMIN": {"LMDA": 0.001}, "MEANTEACHER": {"WEIGHT_U": 1.0, "EMA_ALPHA": 0.999, "RAMPUP": 5}, "MIXMATCH": {"WEIGHT_U": 100.0, "TEMP": 2.0, "MIXUP_BETA": 0.75, "RAMPUP": 20000}, "FIXMATCH": {"WEIGHT_U": 1.0, "CONF_THRE": 0.95, "STRONG_TRANSFORMS": []}, "COOP": {"N_CTX": 16, "CSC": false, "CTX_INIT": "", "PREC": "fp16", "CLASS_TOKEN_POSITION": "end"}, "COCOOP": {"N_CTX": 16, "CTX_INIT": "", "PREC": "fp16"}, "MAPLE": {"N_CTX": 2, "CTX_INIT": "a photo of a", "PREC": "fp16", "PROMPT_DEPTH": 3}, "PROMPTSRC": {"N_CTX_VISION": 4, "N_CTX_TEXT": 4, "CTX_INIT": "a photo of a", "PREC": "fp16", "PROMPT_DEPTH_VISION": 9, "PROMPT_DEPTH_TEXT": 9, "TEXT_LOSS_WEIGHT": 25, "IMAGE_LOSS_WEIGHT": 10, "GPA_MEAN": 15, "GPA_STD": 1}, "IVLP": {"N_CTX_VISION": 2, "N_CTX_TEXT": 2, "CTX_INIT": "a photo of a", "PREC": "fp16", "PROMPT_DEPTH_VISION": 9, "PROMPT_DEPTH_TEXT": 9}}}