diff --git a/examples/pytorch/token-classification/README.md b/examples/pytorch/token-classification/README.md index 02a870c92fd..90ea6317c2e 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 \