From ba37ddc5ced98c97e2e3e21018dda14a84e3d257 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Wed, 6 Feb 2019 00:07:08 +0100 Subject: [PATCH] fix run_lm_modeling example command line --- README.md | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index b124585bbe..68de1aa1aa 100644 --- a/README.md +++ b/README.md @@ -680,14 +680,15 @@ Training one epoch on this corpus takes about 1:20h on 4 x NVIDIA Tesla P100 wit ```shell python run_lm_finetuning.py \ - --bert_model bert-base-cased - --do_train - --train_file samples/sample_text.txt - --output_dir models - --num_train_epochs 5.0 - --learning_rate 3e-5 - --train_batch_size 32 - --max_seq_length 128 + --bert_model bert-base-uncased \ + --do_lower_case \ + --do_train \ + --train_file ../samples/sample_text.txt \ + --output_dir models \ + --num_train_epochs 5.0 \ + --learning_rate 3e-5 \ + --train_batch_size 32 \ + --max_seq_length 128 \ ``` ## Fine-tuning BERT-large on GPUs