itdainb commited on
Commit
5466115
·
verified ·
1 Parent(s): 8cd728d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -4
README.md CHANGED
@@ -34,7 +34,8 @@ pipeline_tag: text-classification
34
  3. [Usage with `sentence-transformers`](#usage-with-sentence-transformers)
35
  4. [Usage with `transformers`](#usage-with-transformers)
36
  5. [Performance](#performance)
37
- 6. [Citation](#citation)
 
38
 
39
 
40
  ## Installation
@@ -100,14 +101,12 @@ model.half()
100
 
101
  features = tokenizer(tokenized_pairs, padding=True, truncation="longest_first", return_tensors="pt", max_length=MAX_LENGTH)
102
 
103
- activation_function = torch.nn.Sigmoid() if model.config.num_labels == 1 else torch.nn.Identity()
104
-
105
  model.eval()
106
  with torch.no_grad():
107
  model_predictions = model(**features, return_dict=True)
108
 
109
  logits = model_predictions.logits
110
- logits = activation_function(logits)
111
  scores = [logit[0] for logit in logits]
112
 
113
  # 0.9819, 0.2444, 0.9253
@@ -127,6 +126,16 @@ In the following table, we provide various pre-trained Cross-Encoders together w
127
 
128
  Note: Runtime was computed on a A100 GPU with fp16.
129
 
 
 
 
 
 
 
 
 
 
 
130
  ## Citation
131
 
132
  Please cite as
 
34
  3. [Usage with `sentence-transformers`](#usage-with-sentence-transformers)
35
  4. [Usage with `transformers`](#usage-with-transformers)
36
  5. [Performance](#performance)
37
+ 6. [Support me](#support-me)
38
+ 7. [Citation](#citation)
39
 
40
 
41
  ## Installation
 
101
 
102
  features = tokenizer(tokenized_pairs, padding=True, truncation="longest_first", return_tensors="pt", max_length=MAX_LENGTH)
103
 
 
 
104
  model.eval()
105
  with torch.no_grad():
106
  model_predictions = model(**features, return_dict=True)
107
 
108
  logits = model_predictions.logits
109
+ logits = torch.nn.Sigmoid()(logits)
110
  scores = [logit[0] for logit in logits]
111
 
112
  # 0.9819, 0.2444, 0.9253
 
126
 
127
  Note: Runtime was computed on a A100 GPU with fp16.
128
 
129
+ ## Support me
130
+ If you find this work useful and would like to support its continued development, here are a few ways you can help:
131
+
132
+ 1. **Star the Repository**: If you appreciate this work, please give it a star. Your support encourages continued development and improvement.
133
+ 2. **Contribute**: Contributions are always welcome! You can help by reporting issues, submitting pull requests, or suggesting new features.
134
+ 3. **Share**: Share this project with your colleagues, friends, or community. The more people know about it, the more feedback and contributions it can attract.
135
+ 4. **Buy me a coffee**: If you’d like to provide financial support, consider making a donation. You can donate via
136
+ - Momo: 0948798843
137
+ - BIDV Bank: DAINB
138
+
139
  ## Citation
140
 
141
  Please cite as