|
|
6 years ago | |
|---|---|---|
| .vscode | 6 years ago | |
| data | 6 years ago | |
| img | 6 years ago | |
| saved_model | 6 years ago | |
| sqlnet | 6 years ago | |
| .dockerignore | 6 years ago | |
| .gitignore | 6 years ago | |
| Dockerfile | 6 years ago | |
| LICENSE | 6 years ago | |
| README.cn.md | 6 years ago | |
| README.md | 6 years ago | |
| docker-compose.debug.yml | 6 years ago | |
| docker-compose.yml | 6 years ago | |
| pip.conf | 6 years ago | |
| requirements.txt | 6 years ago | |
| start_test.sh | 6 years ago | |
| start_train.sh | 6 years ago | |
| test.py | 6 years ago | |
| train.py | 6 years ago |
https://drive.google.com/open?id=10RtAom_D4zOp_w5OsYLTtLd2TEuC7C1X
模型训练所需要的算力还是挺大的,在 K80 上训练一个 Epoch 大概需要 7 到 8 分钟,基线模型默认训练 100 个 Epoch。追一表示试验多次后,验证集上的平均 Logic Form Accuracy 为 27.5%,即 SQL 每一个子句全都预测正确的概率为 27.5%。
WikiSQL 基准:https://github.com/salesforce/WikiSQL
CUDA_VISIBLE_DEVICES=0 python train.py --ca --gpu --bs 128