Update README.md
Browse files
README.md
CHANGED
@@ -14,10 +14,6 @@ You can download trained model:
|
|
14 |
- [tensorflow](https://drive.google.com/file/d/1X-sRDYf7moS_h61J3L79NkMVGHP-P-k5/view?usp=sharing).
|
15 |
- [pytorch](https://drive.google.com/file/d/11aFSTpYIurn-oI2XpAmcCTccB_AonMOu/view?usp=sharing).
|
16 |
|
17 |
-
***************New Mar 11 , 2020 ***************
|
18 |
-
|
19 |
-
**[BERT](https://github.com/google-research/bert)** (from Google Research and the Toyota Technological Institute at Chicago) released with the paper [BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding](https://arxiv.org/abs/1810.04805).
|
20 |
-
|
21 |
Use with huggingface/transformers
|
22 |
``` bash
|
23 |
import torch
|
@@ -100,11 +96,11 @@ Run training with base config
|
|
100 |
|
101 |
``` bash
|
102 |
|
103 |
-
python train_pytorch.py
|
104 |
-
--model_path=bert4news.pytorch
|
105 |
-
--max_len=200
|
106 |
-
--batch_size=16
|
107 |
-
--epochs=6
|
108 |
--lr=2e-5
|
109 |
|
110 |
```
|
|
|
14 |
- [tensorflow](https://drive.google.com/file/d/1X-sRDYf7moS_h61J3L79NkMVGHP-P-k5/view?usp=sharing).
|
15 |
- [pytorch](https://drive.google.com/file/d/11aFSTpYIurn-oI2XpAmcCTccB_AonMOu/view?usp=sharing).
|
16 |
|
|
|
|
|
|
|
|
|
17 |
Use with huggingface/transformers
|
18 |
``` bash
|
19 |
import torch
|
|
|
96 |
|
97 |
``` bash
|
98 |
|
99 |
+
python train_pytorch.py \\\\
|
100 |
+
--model_path=bert4news.pytorch \\\\
|
101 |
+
--max_len=200 \\\\
|
102 |
+
--batch_size=16 \\\\
|
103 |
+
--epochs=6 \\\\
|
104 |
--lr=2e-5
|
105 |
|
106 |
```
|