saiful9379
commited on
Commit
•
30206d4
1
Parent(s):
371d540
Update README.md
Browse files
README.md
CHANGED
@@ -29,4 +29,14 @@ output = model.generate(
|
|
29 |
)
|
30 |
predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
31 |
print(predicted_text)
|
32 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
)
|
30 |
predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
31 |
print(predicted_text)
|
32 |
+
```
|
33 |
+
|
34 |
+
Model configuration,
|
35 |
+
Here is the basic configuration of Bangla GPT2 Model,
|
36 |
+
```
|
37 |
+
vocab_size = 50000
|
38 |
+
block_size = 200
|
39 |
+
learning_rate=3e-5
|
40 |
+
num_epoch = 100
|
41 |
+
batch_size = 12
|
42 |
+
buffer_size = 1000
|