Skip to content

GitLab

Unverified Commit 803a8cd1 authored by Kamal Raj's avatar Kamal Raj Committed by GitHub
Browse files

updated readme with proper arguments (#14624)

parent 3977b584
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
examples/pytorch/token-classification/README.md
View file @ 803a8cd1
......@@ -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 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment