From 803a8cd18f44e7288187b82ae3b48956edf11dd6 Mon Sep 17 00:00:00 2001 From: Kamal Raj Date: Mon, 6 Dec 2021 08:42:51 +0530 Subject: [PATCH] updated readme with proper arguments (#14624) --- examples/pytorch/token-classification/README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/pytorch/token-classification/README.md b/examples/pytorch/token-classification/README.md index 02a870c92f..90ea6317c2 100644 --- a/examples/pytorch/token-classification/README.md +++ b/examples/pytorch/token-classification/README.md @@ -83,8 +83,9 @@ export TASK_NAME=ner python run_ner_no_trainer.py \ --model_name_or_path bert-base-cased \ + --dataset_name conll2003 \ --task_name $TASK_NAME \ - --max_seq_length 128 \ + --max_length 128 \ --per_device_train_batch_size 32 \ --learning_rate 2e-5 \ --num_train_epochs 3 \ @@ -110,8 +111,9 @@ export TASK_NAME=ner accelerate launch run_ner_no_trainer.py \ --model_name_or_path bert-base-cased \ + --dataset_name conll2003 \ --task_name $TASK_NAME \ - --max_seq_length 128 \ + --max_length 128 \ --per_device_train_batch_size 32 \ --learning_rate 2e-5 \ --num_train_epochs 3 \