Iulian277 commited on
Commit
44de8df
1 Parent(s): 6c58c4f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -0
config.json CHANGED
@@ -34,6 +34,16 @@
34
  "LABEL_1": 1,
35
  "LABEL_2": 2
36
  },
 
 
 
 
 
 
 
 
 
 
37
  "max_position_embeddings": 2048,
38
  "model_type": "bart",
39
  "num_hidden_layers": 6,
 
34
  "LABEL_1": 1,
35
  "LABEL_2": 2
36
  },
37
+ "task_specific_params": {
38
+ "summarization": {
39
+ "early_stopping": true,
40
+ "length_penalty": 2.0,
41
+ "max_length": 100,
42
+ "min_length": 10,
43
+ "no_repeat_ngram_size": 3,
44
+ "num_beams": 4
45
+ }
46
+ },
47
  "max_position_embeddings": 2048,
48
  "model_type": "bart",
49
  "num_hidden_layers": 6,