File size: 2,186 Bytes
4424e71 |
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 |
{
"tensorboard": true,
"wandb": true,
"exp_name": "common0526_1_vitb16_TinyImageNet_split4_2",
"cfg": "configs/TinyImageNet/vitb16_common0526_1.yaml",
"n_gpu": 1,
"model_arch": "vitb16",
"checkpoint_path": "runs/train/common0526_1_vitb16_TinyImageNet_split4_1_exp/ckpts/best.ckpt",
"image_size": 224,
"batch_size": 64,
"num_workers_pgpu": 10,
"max_epochs": null,
"warmup_epochs": null,
"train_steps": 4590,
"warmup_steps": 500,
"lr": 0.01,
"lr_strategy": "one_cycle",
"num_restarts": null,
"wd": 0.0001,
"data_dir": "./data",
"dataset": "TinyImageNet",
"num_classes": 20,
"known_classes": [
98,
36,
158,
177,
189,
157,
170,
191,
82,
196,
138,
166,
43,
13,
152,
11,
75,
174,
193,
190
],
"osr_dataset": null,
"osr_classes": null,
"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_1_vitb16_TinyImageNet_split4_2_exp2/tb/",
"checkpoint_dir": "runs/train/common0526_1_vitb16_TinyImageNet_split4_2_exp2/ckpts/",
"embedding_dir": "runs/train/common0526_1_vitb16_TinyImageNet_split4_2_exp2/embeddings/",
"lightning_dir": "runs/train/common0526_1_vitb16_TinyImageNet_split4_2_exp2/lightning_logs/",
"result_dir": "runs/train/common0526_1_vitb16_TinyImageNet_split4_2_exp2"
} |