Update config.json
Browse files- config.json +131 -0
config.json
CHANGED
@@ -115,3 +115,134 @@
|
|
115 |
"use_cache": true,
|
116 |
"vocab_size": 50265
|
117 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
115 |
"use_cache": true,
|
116 |
"vocab_size": 50265
|
117 |
}
|
118 |
+
|
119 |
+
|
120 |
+
|
121 |
+
|
122 |
+
|
123 |
+
|
124 |
+
|
125 |
+
|
126 |
+
|
127 |
+
|
128 |
+
|
129 |
+
|
130 |
+
|
131 |
+
|
132 |
+
{
|
133 |
+
"_name_or_path": "/root/.cache/torch/sentence_transformers/facebook_bart-large-mnli",
|
134 |
+
"_num_labels": 3,
|
135 |
+
"activation_dropout": 0.0,
|
136 |
+
"activation_function": "gelu",
|
137 |
+
"add_final_layer_norm": false,
|
138 |
+
"architectures": [
|
139 |
+
"BartModel"
|
140 |
+
],
|
141 |
+
"attention_dropout": 0.0,
|
142 |
+
"bos_token_id": 0,
|
143 |
+
"classif_dropout": 0.0,
|
144 |
+
"classifier_dropout": 0.0,
|
145 |
+
"d_model": 1024,
|
146 |
+
"decoder_attention_heads": 16,
|
147 |
+
"decoder_ffn_dim": 4096,
|
148 |
+
"decoder_layerdrop": 0.0,
|
149 |
+
"decoder_layers": 12,
|
150 |
+
"decoder_start_token_id": 2,
|
151 |
+
"dropout": 0.1,
|
152 |
+
"encoder_attention_heads": 16,
|
153 |
+
"encoder_ffn_dim": 4096,
|
154 |
+
"encoder_layerdrop": 0.0,
|
155 |
+
"encoder_layers": 12,
|
156 |
+
"eos_token_id": 2,
|
157 |
+
"forced_eos_token_id": 2,
|
158 |
+
"gradient_checkpointing": false,
|
159 |
+
"id2label": {
|
160 |
+
"0": "atmosphere",
|
161 |
+
"1": "beer list",
|
162 |
+
"2": "bottomless brunch",
|
163 |
+
"3": "breakfast",
|
164 |
+
"4": "brunch",
|
165 |
+
"5": "business",
|
166 |
+
"6": "byob",
|
167 |
+
"7": "casual",
|
168 |
+
"8": "date night",
|
169 |
+
"9": "delivery",
|
170 |
+
"10": "fine dining",
|
171 |
+
"11": "gluten free",
|
172 |
+
"12": "good value",
|
173 |
+
"13": "great cocktails",
|
174 |
+
"14": "great service",
|
175 |
+
"15": "happy hour",
|
176 |
+
"16": "healthy",
|
177 |
+
"17": "instagrammable",
|
178 |
+
"18": "kid friendly",
|
179 |
+
"19": "large groups",
|
180 |
+
"20": "late night",
|
181 |
+
"21": "live music",
|
182 |
+
"22": "lunch",
|
183 |
+
"23": "night out",
|
184 |
+
"24": "outdoor seating",
|
185 |
+
"25": "quick",
|
186 |
+
"26": "romantic",
|
187 |
+
"27": "rooftop",
|
188 |
+
"28": "small plates",
|
189 |
+
"29": "special event",
|
190 |
+
"30": "sports/tvs",
|
191 |
+
"31": "takeout",
|
192 |
+
"32": "tasting menu",
|
193 |
+
"33": "vegetarian",
|
194 |
+
"34": "walk-in",
|
195 |
+
"35": "wine list"
|
196 |
+
},
|
197 |
+
"init_std": 0.02,
|
198 |
+
"is_encoder_decoder": true,
|
199 |
+
"label2id": {
|
200 |
+
"atmosphere": 0,
|
201 |
+
"beer list": 1,
|
202 |
+
"bottomless brunch": 2,
|
203 |
+
"breakfast": 3,
|
204 |
+
"brunch": 4,
|
205 |
+
"business": 5,
|
206 |
+
"byob": 6,
|
207 |
+
"casual": 7,
|
208 |
+
"date night": 8,
|
209 |
+
"delivery": 9,
|
210 |
+
"fine dining": 10,
|
211 |
+
"gluten free": 11,
|
212 |
+
"good value": 12,
|
213 |
+
"great cocktails": 13,
|
214 |
+
"great service": 14,
|
215 |
+
"happy hour": 15,
|
216 |
+
"healthy": 16,
|
217 |
+
"instagrammable": 17,
|
218 |
+
"kid friendly": 18,
|
219 |
+
"large groups": 19,
|
220 |
+
"late night": 20,
|
221 |
+
"live music": 21,
|
222 |
+
"lunch": 22,
|
223 |
+
"night out": 23,
|
224 |
+
"outdoor seating": 24,
|
225 |
+
"quick": 25,
|
226 |
+
"romantic": 26,
|
227 |
+
"rooftop": 27,
|
228 |
+
"small plates": 28,
|
229 |
+
"special event": 29,
|
230 |
+
"sports/tvs": 30,
|
231 |
+
"takeout": 31,
|
232 |
+
"tasting menu": 32,
|
233 |
+
"vegetarian": 33,
|
234 |
+
"walk-in": 34,
|
235 |
+
"wine list": 35
|
236 |
+
},
|
237 |
+
"max_position_embeddings": 1024,
|
238 |
+
"model_type": "bart",
|
239 |
+
"normalize_before": false,
|
240 |
+
"num_hidden_layers": 12,
|
241 |
+
"output_past": false,
|
242 |
+
"pad_token_id": 1,
|
243 |
+
"scale_embedding": false,
|
244 |
+
"torch_dtype": "float32",
|
245 |
+
"transformers_version": "4.26.0",
|
246 |
+
"use_cache": true,
|
247 |
+
"vocab_size": 50265
|
248 |
+
}
|