common_cnnr101_CIFAR10+ood_split1_2_exp/config.json

{
    "tensorboard": true,
    "wandb": true,
    "exp_name": "common0610_3_cnnr101_CIFAR10+ood_split1_2",
    "cfg": "configs/CIFAR10+ood/cnnr101_common0610_3.yaml",
    "random_seed": 2023,
    "n_gpu": 1,
    "model_arch": "cnnr101",
    "image_size": 128,
    "checkpoint_path": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_1_exp/ckpts/best.ckpt",
    "data_dir": "./data",
    "dataset": "CIFAR10",
    "num_classes": 10,
    "known_classes": [
        0,
        1,
        2,
        3,
        4,
        5,
        6,
        7,
        8,
        9
    ],
    "osr_dataset": "ImageNetCrop",
    "osr_classes": [
        0,
        1,
        2,
        3,
        4,
        5,
        6,
        7,
        8,
        9,
        10,
        11,
        12,
        13,
        14,
        15,
        16,
        17,
        18,
        19,
        20,
        21,
        22,
        23,
        24,
        25,
        26,
        27,
        28,
        29,
        30,
        31,
        32,
        33,
        34,
        35,
        36,
        37,
        38,
        39,
        40,
        41,
        42,
        43,
        44,
        45,
        46,
        47,
        48,
        49,
        50,
        51,
        52,
        53,
        54,
        55,
        56,
        57,
        58,
        59,
        60,
        61,
        62,
        63,
        64,
        65,
        66,
        67,
        68,
        69,
        70,
        71,
        72,
        73,
        74,
        75,
        76,
        77,
        78,
        79,
        80,
        81,
        82,
        83,
        84,
        85,
        86,
        87,
        88,
        89,
        90,
        91,
        92,
        93,
        94,
        95,
        96,
        97,
        98,
        99,
        100,
        101,
        102,
        103,
        104,
        105,
        106,
        107,
        108,
        109,
        110,
        111,
        112,
        113,
        114,
        115,
        116,
        117,
        118,
        119,
        120,
        121,
        122,
        123,
        124,
        125,
        126,
        127,
        128,
        129,
        130,
        131,
        132,
        133,
        134,
        135,
        136,
        137,
        138,
        139,
        140,
        141,
        142,
        143,
        144,
        145,
        146,
        147,
        148,
        149,
        150,
        151,
        152,
        153,
        154,
        155,
        156,
        157,
        158,
        159,
        160,
        161,
        162,
        163,
        164,
        165,
        166,
        167,
        168,
        169,
        170,
        171,
        172,
        173,
        174,
        175,
        176,
        177,
        178,
        179,
        180,
        181,
        182,
        183,
        184,
        185,
        186,
        187,
        188,
        189,
        190,
        191,
        192,
        193,
        194,
        195,
        196,
        197,
        198,
        199
    ],
    "batch_size": 254,
    "num_workers_pgpu": 10,
    "max_epochs": 100,
    "warmup_epochs": 10,
    "train_steps": null,
    "warmup_steps": null,
    "lr": 0.0001,
    "lr_strategy": "cosine",
    "num_restarts": 2,
    "wd": 0.0001,
    "eval": false,
    "opt": "SGD",
    "save_freq": 50,
    "label_smoothing": 0,
    "transform": "default",
    "sim_block": "attention",
    "num_sim_block": 1,
    "sim_task": "reg",
    "plot_model": false,
    "plot_confusion_matrix": false,
    "plot_feature_embedding": false,
    "plot_roc": false,
    "sim_alpha": 0.01,
    "class_anchor": "mean",
    "anchor_freq": 10,
    "fusion": "cross-attention",
    "model_config": {
        "sim_block": "attention",
        "num_sim_block": 1,
        "sim_task": "reg",
        "emb_dim": 768,
        "mlp_dim": 3072,
        "sim_num_heads": 12,
        "num_heads": 12,
        "num_layers": 12,
        "attn_dropout_rate": 0.0,
        "dropout_rate": 0.1
    },
    "tensorboard_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/tb/",
    "checkpoint_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/ckpts/",
    "embedding_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/embeddings/",
    "lightning_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/lightning_logs/",
    "result_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp"
}