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

Upload 2 files

Browse files
common_cnnr101_CIFAR10+ood_split1_2_exp/ckpts/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c09442f3431d054fd9a1856b76eac01124aa6a4b5b64732c8a3454e8fc0d999b
3
+ size 556583324
common_cnnr101_CIFAR10+ood_split1_2_exp/config.json ADDED
@@ -0,0 +1,272 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tensorboard": true,
3
+ "wandb": true,
4
+ "exp_name": "common0610_3_cnnr101_CIFAR10+ood_split1_2",
5
+ "cfg": "configs/CIFAR10+ood/cnnr101_common0610_3.yaml",
6
+ "random_seed": 2023,
7
+ "n_gpu": 1,
8
+ "model_arch": "cnnr101",
9
+ "image_size": 128,
10
+ "checkpoint_path": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_1_exp/ckpts/best.ckpt",
11
+ "data_dir": "./data",
12
+ "dataset": "CIFAR10",
13
+ "num_classes": 10,
14
+ "known_classes": [
15
+ 0,
16
+ 1,
17
+ 2,
18
+ 3,
19
+ 4,
20
+ 5,
21
+ 6,
22
+ 7,
23
+ 8,
24
+ 9
25
+ ],
26
+ "osr_dataset": "ImageNetCrop",
27
+ "osr_classes": [
28
+ 0,
29
+ 1,
30
+ 2,
31
+ 3,
32
+ 4,
33
+ 5,
34
+ 6,
35
+ 7,
36
+ 8,
37
+ 9,
38
+ 10,
39
+ 11,
40
+ 12,
41
+ 13,
42
+ 14,
43
+ 15,
44
+ 16,
45
+ 17,
46
+ 18,
47
+ 19,
48
+ 20,
49
+ 21,
50
+ 22,
51
+ 23,
52
+ 24,
53
+ 25,
54
+ 26,
55
+ 27,
56
+ 28,
57
+ 29,
58
+ 30,
59
+ 31,
60
+ 32,
61
+ 33,
62
+ 34,
63
+ 35,
64
+ 36,
65
+ 37,
66
+ 38,
67
+ 39,
68
+ 40,
69
+ 41,
70
+ 42,
71
+ 43,
72
+ 44,
73
+ 45,
74
+ 46,
75
+ 47,
76
+ 48,
77
+ 49,
78
+ 50,
79
+ 51,
80
+ 52,
81
+ 53,
82
+ 54,
83
+ 55,
84
+ 56,
85
+ 57,
86
+ 58,
87
+ 59,
88
+ 60,
89
+ 61,
90
+ 62,
91
+ 63,
92
+ 64,
93
+ 65,
94
+ 66,
95
+ 67,
96
+ 68,
97
+ 69,
98
+ 70,
99
+ 71,
100
+ 72,
101
+ 73,
102
+ 74,
103
+ 75,
104
+ 76,
105
+ 77,
106
+ 78,
107
+ 79,
108
+ 80,
109
+ 81,
110
+ 82,
111
+ 83,
112
+ 84,
113
+ 85,
114
+ 86,
115
+ 87,
116
+ 88,
117
+ 89,
118
+ 90,
119
+ 91,
120
+ 92,
121
+ 93,
122
+ 94,
123
+ 95,
124
+ 96,
125
+ 97,
126
+ 98,
127
+ 99,
128
+ 100,
129
+ 101,
130
+ 102,
131
+ 103,
132
+ 104,
133
+ 105,
134
+ 106,
135
+ 107,
136
+ 108,
137
+ 109,
138
+ 110,
139
+ 111,
140
+ 112,
141
+ 113,
142
+ 114,
143
+ 115,
144
+ 116,
145
+ 117,
146
+ 118,
147
+ 119,
148
+ 120,
149
+ 121,
150
+ 122,
151
+ 123,
152
+ 124,
153
+ 125,
154
+ 126,
155
+ 127,
156
+ 128,
157
+ 129,
158
+ 130,
159
+ 131,
160
+ 132,
161
+ 133,
162
+ 134,
163
+ 135,
164
+ 136,
165
+ 137,
166
+ 138,
167
+ 139,
168
+ 140,
169
+ 141,
170
+ 142,
171
+ 143,
172
+ 144,
173
+ 145,
174
+ 146,
175
+ 147,
176
+ 148,
177
+ 149,
178
+ 150,
179
+ 151,
180
+ 152,
181
+ 153,
182
+ 154,
183
+ 155,
184
+ 156,
185
+ 157,
186
+ 158,
187
+ 159,
188
+ 160,
189
+ 161,
190
+ 162,
191
+ 163,
192
+ 164,
193
+ 165,
194
+ 166,
195
+ 167,
196
+ 168,
197
+ 169,
198
+ 170,
199
+ 171,
200
+ 172,
201
+ 173,
202
+ 174,
203
+ 175,
204
+ 176,
205
+ 177,
206
+ 178,
207
+ 179,
208
+ 180,
209
+ 181,
210
+ 182,
211
+ 183,
212
+ 184,
213
+ 185,
214
+ 186,
215
+ 187,
216
+ 188,
217
+ 189,
218
+ 190,
219
+ 191,
220
+ 192,
221
+ 193,
222
+ 194,
223
+ 195,
224
+ 196,
225
+ 197,
226
+ 198,
227
+ 199
228
+ ],
229
+ "batch_size": 254,
230
+ "num_workers_pgpu": 10,
231
+ "max_epochs": 100,
232
+ "warmup_epochs": 10,
233
+ "train_steps": null,
234
+ "warmup_steps": null,
235
+ "lr": 0.0001,
236
+ "lr_strategy": "cosine",
237
+ "num_restarts": 2,
238
+ "wd": 0.0001,
239
+ "eval": false,
240
+ "opt": "SGD",
241
+ "save_freq": 50,
242
+ "label_smoothing": 0,
243
+ "transform": "default",
244
+ "sim_block": "attention",
245
+ "num_sim_block": 1,
246
+ "sim_task": "reg",
247
+ "plot_model": false,
248
+ "plot_confusion_matrix": false,
249
+ "plot_feature_embedding": false,
250
+ "plot_roc": false,
251
+ "sim_alpha": 0.01,
252
+ "class_anchor": "mean",
253
+ "anchor_freq": 10,
254
+ "fusion": "cross-attention",
255
+ "model_config": {
256
+ "sim_block": "attention",
257
+ "num_sim_block": 1,
258
+ "sim_task": "reg",
259
+ "emb_dim": 768,
260
+ "mlp_dim": 3072,
261
+ "sim_num_heads": 12,
262
+ "num_heads": 12,
263
+ "num_layers": 12,
264
+ "attn_dropout_rate": 0.0,
265
+ "dropout_rate": 0.1
266
+ },
267
+ "tensorboard_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/tb/",
268
+ "checkpoint_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/ckpts/",
269
+ "embedding_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/embeddings/",
270
+ "lightning_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp/lightning_logs/",
271
+ "result_dir": "runs/train/common0610_3_cnnr101_CIFAR10+ood_split1_2_exp"
272
+ }