softpe_20.json

{
    "TUNING" : "softcpt",
    "MODEL"  : "pecore",
    "TASK"   : -1,
    "MODEL_TYPE" : "PE-Core-L14-336",

    "NUM_VISUAL_PROMPT" : 0,
    "NUM_TEXT_CNTX" : 20,
    "TASK_NAMES": ["age estimation from face picture", "gender recognition from facial features", "emotion classification from facial expression"], 
    "CLASSES": [
        ["0-2", "3-9", "10-19", "20-29", "30-39", "40-49", "50-59", "60-69", "70+"],
        ["male", "female"],
        ["surprise", "fear", "disgust", "happy", "sad", "angry", "neutral"]
    ],    
    "NAMED_TRAINABLE_PARAMETERS": [
        "prompt_learner",
        "task_prompt_learner", 
        "prompt_gen"
    ],

    "DATASET_NAMES": {
        "-1" : ["FairFace", "RAF-DB", "CelebA_HQ", "Lagenda"] 
    },
    "CSP": false,
    "EMD_WEIGHT" : 30,
    "EMD_OMEGA" : 2.0,
    "EMD_MU": -0.0025,
    "DATASET_ROOT": "../processed_datasets/datasets_with_standard_labels",
    "BALANCE_TASK" : {"2" : 0.33},
    "BATCH_SIZE" : 256,
    "NUM_WORKERS" : 3,
    "LR" : 0.002,
    "OUTPUT_DIR" : "../TRAIN/PECore/L14/SoftCPT/TSCA_cntx_20",
    "PREFETCH_FACTOR" : 1,
    "VERBOSE": true,
    "USE_TQDM": false,
    "EPOCHS" : 50,
    "PATIENCE" : 7

}