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-
|