Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,10 @@ train data is similarity sentence data from E-commerce dialogue, about 50w sente
|
|
12 |
## Model
|
13 |
model created by [sentence-tansformers](https://www.sbert.net/index.html),model struct is cross-encoder,pretrained model is hfl/chinese-roberta-wwm-ext-large.
|
14 |
|
15 |
-
###
|
|
|
|
|
|
|
16 |
```python
|
17 |
>>> from sentence_transformers.cross_encoder import CrossEncoder
|
18 |
>>> model = CrossEncoder(model_save_path, device="cuda", max_length=64)
|
|
|
12 |
## Model
|
13 |
model created by [sentence-tansformers](https://www.sbert.net/index.html),model struct is cross-encoder,pretrained model is hfl/chinese-roberta-wwm-ext-large.
|
14 |
|
15 |
+
###Code
|
16 |
+
train code from https://github.com/TTurn/cross-encoder
|
17 |
+
|
18 |
+
#### Usage
|
19 |
```python
|
20 |
>>> from sentence_transformers.cross_encoder import CrossEncoder
|
21 |
>>> model = CrossEncoder(model_save_path, device="cuda", max_length=64)
|