Sihao Chen commited on
Commit
afa3f84
1 Parent(s): e25ec19

Update number of labels for the model

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "facebook/bart-base",
3
  "activation_dropout": 0.1,
4
  "activation_function": "gelu",
5
  "add_bias_logits": false,
 
1
  {
2
+ "_num_labels": 3,
3
  "activation_dropout": 0.1,
4
  "activation_function": "gelu",
5
  "add_bias_logits": false,