Updated README.md
Browse files
README.md
CHANGED
|
@@ -4,10 +4,13 @@ The model was trained on the Spider dataset Link: https://yale-lily.github.io/sp
|
|
| 4 |
The model was trained using Google colab.
|
| 5 |
|
| 6 |
Hyperparameters:
|
| 7 |
-
"
|
| 8 |
"learning rate" = 1e-5
|
| 9 |
-
"
|
| 10 |
-
"
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
-
"model name" : "facebook/bart-large-cnn"
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
The model was trained using Google colab.
|
| 5 |
|
| 6 |
Hyperparameters:
|
| 7 |
+
"num epochs" = 3
|
| 8 |
"learning rate" = 1e-5
|
| 9 |
+
"batch size" = 8
|
| 10 |
+
"weight decay" = 0.01
|
| 11 |
+
"max input length" = 256
|
| 12 |
+
"max target length" = 256
|
| 13 |
+
"model name" : "facebook/bart-large-cnn"
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
Link to the github repo containing training notebook: https://github.com/vanadnarayane26/Text_to_SQL_Spider-
|