Update train.yaml
This commit is contained in:
parent
fceb7b5525
commit
606b2a9bc7
|
@ -1,25 +1,25 @@
|
|||
data_dir: "data/"
|
||||
bert_model: "bert-base-chinese"
|
||||
task_name: "ner"
|
||||
output_dir: "checkpoints"
|
||||
max_seq_length: 128
|
||||
do_train: True
|
||||
do_eval: True
|
||||
eval_on: "dev"
|
||||
do_lower_case: True
|
||||
train_batch_size: 32
|
||||
eval_batch_size: 8
|
||||
learning_rate: 5e-5
|
||||
num_train_epochs: 3 # the number of training epochs
|
||||
warmup_proportion: 0.1
|
||||
weight_decay: 0.01
|
||||
adam_epsilon: 1e-8
|
||||
max_grad_norm: 1.0
|
||||
use_gpu: True # use gpu or not
|
||||
gpu_id: 1
|
||||
local_rank: -1
|
||||
seed: 42
|
||||
gradient_accumulation_steps: 1
|
||||
bert_model: "bert-base-chinese"
|
||||
data_dir: "data/"
|
||||
do_eval: True
|
||||
do_lower_case: True
|
||||
do_train: True
|
||||
eval_batch_size: 8
|
||||
eval_on: "dev"
|
||||
fp16: False
|
||||
fp16_opt_level: "01"
|
||||
gpu_id: 1
|
||||
gradient_accumulation_steps: 1
|
||||
learning_rate: 5e-5
|
||||
local_rank: -1
|
||||
loss_scale: 0.0
|
||||
max_grad_norm: 1.0
|
||||
max_seq_length: 128
|
||||
num_train_epochs: 3 # the number of training epochs
|
||||
output_dir: "checkpoints"
|
||||
seed: 42
|
||||
task_name: "ner"
|
||||
train_batch_size: 32
|
||||
use_gpu: True # use gpu or not
|
||||
warmup_proportion: 0.1
|
||||
weight_decay: 0.01
|
||||
|
|
Loading…
Reference in New Issue