diegoale1248 commited on
Commit
f2f4ffb
1 Parent(s): 424e6cf

manual update

Browse files
Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "neuralmind/bert-base-portuguese-cased",
3
  "architectures": [
4
- "BertForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
@@ -374,6 +374,8 @@
374
  "pooler_size_per_head": 128,
375
  "pooler_type": "first_token_transform",
376
  "position_embedding_type": "absolute",
 
 
377
  "transformers_version": "4.35.2",
378
  "type_vocab_size": 2,
379
  "use_cache": true,
 
1
  {
2
  "_name_or_path": "neuralmind/bert-base-portuguese-cased",
3
  "architectures": [
4
+ "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
 
374
  "pooler_size_per_head": 128,
375
  "pooler_type": "first_token_transform",
376
  "position_embedding_type": "absolute",
377
+ "problem_type": "single_label_classification",
378
+ "torch_dtype": "float32",
379
  "transformers_version": "4.35.2",
380
  "type_vocab_size": 2,
381
  "use_cache": true,