edur0 commited on
Commit
e7a34f9
·
verified ·
1 Parent(s): 0b62184

Add custom classifier

Browse files
Files changed (2) hide show
  1. config.json +2 -2
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "_name_or_path": "BSC-NLP4BIA/test_model",
3
  "architectures": [
4
- "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
 
1
  {
2
+ "_name_or_path": "bert-base-uncased",
3
  "architectures": [
4
+ "CustomBertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cc6bbe5e372186b20bc1e76c2346d4eb00b22f54cc888862ddc4f592f6ddebd7
3
- size 437983256
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a47134764416f3873d82e4f86ea12b804d3210afb2775b41800cb496b584526d
3
+ size 438750416