File size: 455 Bytes
656848b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#!/bin/bash
PLM_DIR=/home/shared/bert-base-uncased-swahili
MODEL_DIR=/home/shared/bert-swahili-news-classification
TASK_NAME=swahili_news
echo "PLM_DIR: ${PLM_DIR}"
echo "MODEL_DIR: ${MODEL_DIR}"
echo "TASK_NAME: ${TASK_NAME}"
python run_flax_glue.py \
--model_name_or_path ${PLM_DIR} \
--task_name ${TASK_NAME} \
--max_length 512 \
--learning_rate 2e-5 \
--num_train_epochs 3 \
--per_device_train_batch_size 4 \
--output_dir ${MODEL_DIR}
|