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
|