Skip to content

Commit fea1532

Browse files
committed
fix training parameters doc
1 parent b9a94ed commit fea1532

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

docs/training_parameters.md

+8-1
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ right time. For reproducibility reasons, we are providing further
3434
skip_correct: 1
3535
skip_complex: 0
3636
max_len: 50
37-
min_len: 3
3837
batch_size: 64
3938
tag_strategy: keep_one
4039
cold_steps_count: 0
@@ -50,6 +49,12 @@ right time. For reproducibility reasons, we are providing further
5049

5150
### Model specific parameters
5251

52+
#### BERT:
53+
```
54+
transformer_model: bert
55+
special_tokens_fix: 0
56+
```
57+
5358
#### XLNet:
5459
```
5560
transformer_model: xlnet
@@ -81,6 +86,7 @@ right time. For reproducibility reasons, we are providing further
8186
updates_per_epoch: 0
8287
tn_prob: 0
8388
tp_prob: 1
89+
pretrain_folder: FOLDER_OF_BEST_MODEL_FROM_STAGE1
8490
pretrain: BEST_MODEL_FROM_STAGE1
8591
```
8692

@@ -101,6 +107,7 @@ right time. For reproducibility reasons, we are providing further
101107
updates_per_epoch: 0
102108
tn_prob: 1
103109
tp_prob: 1
110+
pretrain_folder: FOLDER_OF_BEST_MODEL_FROM_STAGE2
104111
pretrain: BEST_MODEL_FROM_STAGE2
105112
```
106113

0 commit comments

Comments
 (0)