josh-oo commited on
Commit
883b660
·
1 Parent(s): 1514e21

Negative sampling training + contrastive loss

Browse files
Files changed (2) hide show
  1. config.json +3 -3
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "test",
3
  "architectures": [
4
  "BertForSequenceClassificationAdapters"
5
  ],
@@ -16,14 +16,14 @@
16
  "max_position_embeddings": 512,
17
  "model_type": "bert",
18
  "num_attention_heads": 12,
19
- "num_classification_heads": 2,
20
  "num_hidden_layers": 5,
21
  "num_users": 2062,
22
  "pad_token_id": 0,
23
  "position_embedding_type": "absolute",
24
  "problem_type": "single_label_classification",
25
  "torch_dtype": "float32",
26
- "transformers_version": "4.30.1",
27
  "type_vocab_size": 2,
28
  "use_cache": true,
29
  "user_dropout_prob": 0.3,
 
1
  {
2
+ "_name_or_path": "sentence-transformers/all-MiniLM-L6-v2",
3
  "architectures": [
4
  "BertForSequenceClassificationAdapters"
5
  ],
 
16
  "max_position_embeddings": 512,
17
  "model_type": "bert",
18
  "num_attention_heads": 12,
19
+ "num_classification_heads": 1,
20
  "num_hidden_layers": 5,
21
  "num_users": 2062,
22
  "pad_token_id": 0,
23
  "position_embedding_type": "absolute",
24
  "problem_type": "single_label_classification",
25
  "torch_dtype": "float32",
26
+ "transformers_version": "4.30.2",
27
  "type_vocab_size": 2,
28
  "use_cache": true,
29
  "user_dropout_prob": 0.3,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2008e414861c0fd307b0f66fffa4aea609d0affc520014a3de6bce1f0f11180e
3
- size 91270233
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2d895a5d0bf8b4ded80e1b736711c3daa48efa83660d34be39d2566647f5be3
3
+ size 99606105