bert.yaml 532 字节
Newer Older
X
xyzhou-puck 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
bert_config_path: "./config/bert_config.json"
init_checkpoint: None
init_pretraining_params: None
checkpoints: "./saved_model"
epoch: 3
learning_rate: 0.0001
lr_scheduler: "linear_warmup_decay"
weight_decay: 0.01
warmup_proportion: 0.1
save_steps: 100000
validation_steps: 100000
loss_scaling: 1.0
skip_steps: 100
data_dir: None
vocab_path: None
max_seq_len: 512
batch_size: 32
in_tokens: False
do_lower_case: True
random_seed: 5512
X
xyzhou-puck 已提交
21
use_cuda: True
X
xyzhou-puck 已提交
22 23 24 25 26 27
shuffle: True
do_train: True
do_test: True
use_data_parallel: False
verbose: False