File size: 690 Bytes
3124aa4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
set -e
# train schema item classifier
python -u schema_item_classifier.py \
--batch_size 8 \
--gradient_descent_step 2 \
--device "0" \
--learning_rate 1e-5 \
--gamma 2.0 \
--alpha 0.75 \
--epochs 32 \
--patience 16 \
--seed 42 \
--save_path "./models/text2sql_schema_item_classifier_semantic" \
--tensorboard_save_path "./tensorboard_log/text2sql_schema_item_classifier_semantic" \
--train_filepath "./data/preprocessed_data/preprocessed_train_spider_amr.json" \
--dev_filepath "./data/preprocessed_data/preprocessed_dev_amr.json" \
--model_name_or_path "roberta-large" \
--use_contents \
--add_fk_info \
--mode "train"
|