lukecq commited on
Commit
e049916
·
1 Parent(s): 76a49b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -23,6 +23,18 @@ models such as RoBERTa and ALBERT are employed as the backbone, and an output la
23
  The learning objective for FSP is to predict the index of the correct label.
24
  A cross-entropy loss is used for tuning the model.
25
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  ## Intended uses & limitations
27
  The model can be used for zero-shot text classification such sentiment analysis and topic classificaion. No further finetuning is needed.
28
 
 
23
  The learning objective for FSP is to predict the index of the correct label.
24
  A cross-entropy loss is used for tuning the model.
25
 
26
+ ## Model variations
27
+ There are three versions of models released. The details are:
28
+
29
+ | Model | Backbone | #params | accuracy | Speed |
30
+ |------------|-----------|----------|-------|-------|
31
+ | [zero-shot-classify-SSTuning-base](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-base) | [roberta-base)](https://huggingface.co/roberta-base) | 125M | Low | High |
32
+ | [zero-shot-classify-SSTuning-large](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-large) | [roberta-large)](https://huggingface.co/roberta-large) | 355M | Medium | Medium |
33
+ | [zero-shot-classify-SSTuning-ALBERT](https://huggingface.co/DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT) | [albert-xxlarge-v2](https://huggingface.co/albert-xxlarge-v2) | 235M | High | Low |
34
+
35
+
36
+
37
+
38
  ## Intended uses & limitations
39
  The model can be used for zero-shot text classification such sentiment analysis and topic classificaion. No further finetuning is needed.
40