Jankiny commited on
Commit
4b314ef
·
verified ·
1 Parent(s): 6fcb242

Upload 2 files

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