-
Notifications
You must be signed in to change notification settings - Fork 33
/
msra.yaml
49 lines (42 loc) · 852 Bytes
/
msra.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
experiment:
exp_dir: experiments/
exp_name: msra
task: word-segmentation
dataset:
name: msra_cws
namespace: izhx404
preprocessor:
type: sequence-labeling-preprocessor
model_dir: babert-base
max_length: 256
bio2bioes: true
data_collator: SequenceLabelingDataCollatorWithPadding
model:
type: sequence-labeling-model
embedder:
model_name_or_path: babert-base
dropout: 0.1
use_crf: true
train:
max_epochs: 30
dataloader:
batch_size_per_gpu: 32
optimizer:
type: AdamW
lr: 2.0e-5
param_groups:
- regex: crf
lr: 2.0e-5
lr_scheduler:
type: LinearLR
start_factor: 1.0
end_factor: 0.0
total_iters: 30
evaluation:
dataloader:
batch_size_per_gpu: 128
metrics:
- type: ner-metric
- type: ner-dumper
model_type: sequence_labeling
dump_format: conll