vvn's picture
Updated README.md
b4388d7
BART Large CNN model trained for converting NLP queries to SQL queries.
The model was trained on the Spider dataset Link: https://yale-lily.github.io/spider
The model was trained using Google colab.
Hyperparameters:
"num epochs" = 3
"learning rate" = 1e-5
"batch size" = 8
"weight decay" = 0.01
"max input length" = 256
"max target length" = 256
"model name" : "facebook/bart-large-cnn"
Link to the github repo containing training notebook: https://github.com/vanadnarayane26/Text_to_SQL_Spider-