train_text2sql_schema_item_classifier.sh

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"