gangyeolkim
commited on
Commit
•
4da59a8
1
Parent(s):
362e5fe
Upload BartForConditionalGeneration
Browse files- config.json +2 -6
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "gelu",
|
5 |
"add_bias_logits": false,
|
@@ -8,7 +8,7 @@
|
|
8 |
"BartForConditionalGeneration"
|
9 |
],
|
10 |
"attention_dropout": 0.0,
|
11 |
-
"author": "
|
12 |
"bos_token_id": 1,
|
13 |
"classif_dropout": 0.1,
|
14 |
"classifier_dropout": 0.1,
|
@@ -40,14 +40,10 @@
|
|
40 |
"NEGATIVE": 0,
|
41 |
"POSITIVE": 1
|
42 |
},
|
43 |
-
"length_penalty": 2.0,
|
44 |
-
"max_length": 130,
|
45 |
"max_position_embeddings": 1026,
|
46 |
-
"min_length": 30,
|
47 |
"model_type": "bart",
|
48 |
"normalize_before": false,
|
49 |
"normalize_embedding": true,
|
50 |
-
"num_beams": 4,
|
51 |
"num_hidden_layers": 6,
|
52 |
"pad_token_id": 3,
|
53 |
"scale_embedding": false,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "../notebooks/results/korean-total-merged/checkpoint-34000",
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "gelu",
|
5 |
"add_bias_logits": false,
|
|
|
8 |
"BartForConditionalGeneration"
|
9 |
],
|
10 |
"attention_dropout": 0.0,
|
11 |
+
"author": "Heewon Jeon(madjakarta@gmail.com)",
|
12 |
"bos_token_id": 1,
|
13 |
"classif_dropout": 0.1,
|
14 |
"classifier_dropout": 0.1,
|
|
|
40 |
"NEGATIVE": 0,
|
41 |
"POSITIVE": 1
|
42 |
},
|
|
|
|
|
43 |
"max_position_embeddings": 1026,
|
|
|
44 |
"model_type": "bart",
|
45 |
"normalize_before": false,
|
46 |
"normalize_embedding": true,
|
|
|
47 |
"num_hidden_layers": 6,
|
48 |
"pad_token_id": 3,
|
49 |
"scale_embedding": false,
|