Update train.yaml

This commit is contained in:
Xin Xu 2021-11-23 11:16:17 +08:00 committed by GitHub
parent fceb7b5525
commit 606b2a9bc7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 21 deletions

View File

@ -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