BERT_WSD / train.sh
spyzvarun's picture
Upload with huggingface_hub
f9713bb
raw
history blame contribute delete
510 Bytes
python script/run_model.py \
--do_train \
--train_path "data/train/semcor-max_num_gloss=6-augmented.csv" \
--evaluate_during_training \
--eval_path "data/dev/semeval2007.csv" \
--model_name_or_path "bert-base-uncased" \
--output_dir "model/bert_base-augmented-batch_size=128-lr=2e-5-max_gloss=6" \
--save_steps "1000" \
--logging_steps "1000" \
--num_train_epochs "4" \
--per_gpu_train_batch_size "8" \
--gradient_accumulation_steps "16" \
--learning_rate "2e-5"