File size: 2,079 Bytes
4bbcd57 |
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 |
{
"tensorboard": true,
"wandb": true,
"exp_name": "common0526_3_vitb16_CIFAR10+10_split4_2",
"cfg": "configs/CIFAR10+10/vitb16_common0526_3.yaml",
"n_gpu": 1,
"model_arch": "vitb16",
"checkpoint_path": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_1_exp/ckpts/best.ckpt",
"image_size": 128,
"batch_size": 256,
"num_workers_pgpu": 10,
"max_epochs": null,
"warmup_epochs": null,
"train_steps": 4590,
"warmup_steps": 500,
"lr": 0.005,
"lr_strategy": "one_cycle",
"num_restarts": null,
"wd": 0.0001,
"data_dir": "./data",
"dataset": "CIFAR10",
"num_classes": 4,
"known_classes": [
6,
4,
9,
1
],
"osr_dataset": "CIFAR100",
"osr_classes": [
95,
60,
30,
73,
46,
49,
68,
99,
8,
71
],
"random_seed": 2023,
"eval": false,
"opt": "SGD",
"save_freq": 50,
"label_smoothing": 0,
"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",
"patch_size": 16,
"emb_dim": 768,
"mlp_dim": 3072,
"num_heads": 12,
"sim_num_heads": 12,
"num_layers": 12,
"attn_dropout_rate": 0.0,
"dropout_rate": 0.0
},
"tensorboard_dir": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_2_exp/tb/",
"checkpoint_dir": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_2_exp/ckpts/",
"embedding_dir": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_2_exp/embeddings/",
"lightning_dir": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_2_exp/lightning_logs/",
"result_dir": "runs/train/common0526_3_vitb16_CIFAR10+10_split4_2_exp"
} |