Precise task in the readme
Browse files
README.md
CHANGED
@@ -8,6 +8,12 @@ datasets:
|
|
8 |
- wikipedia
|
9 |
---
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
# BERT base model (uncased)
|
12 |
|
13 |
Pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in
|
|
|
8 |
- wikipedia
|
9 |
---
|
10 |
|
11 |
+
This model is exported for masked-lm task with the following command:
|
12 |
+
```
|
13 |
+
python3 -m optimum.exporters.onnx --model bert-base-cased --for-ort --task masked-lm models/
|
14 |
+
```
|
15 |
+
If you want to use `bert-base-uncased` for other tasks, please export the ONNX model with your corresponding task.
|
16 |
+
|
17 |
# BERT base model (uncased)
|
18 |
|
19 |
Pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in
|