|
set -e |
|
|
|
|
|
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" |
|
|