- Python 3.5
- PyTorch 0.2.0.post3 (GPU)
pip install -r requirements.txt
Download the zip file from Google Drive, and copy it to the root folder.
unzip acl18coarse2fine_data_model.zip
./pretrain.sh [geoqueries|atis|django|wikisql] GPU_ID
./run.sh [geoqueries|atis|django|wikisql] GPU_ID
- The implementation is based on OpenNMT/OpenNMT-py.
- The preprocessing and evaluation code used for WikiSQL is from salesforce/WikiSQL.