Update train.py
Browse files
train.py
CHANGED
|
@@ -11,7 +11,7 @@ from util import generate_text
|
|
| 11 |
# ============================
|
| 12 |
# Configuration Parameters
|
| 13 |
# ============================
|
| 14 |
-
dataset_path = 'flpelerin/tinystories-
|
| 15 |
|
| 16 |
num_epochs = 1
|
| 17 |
batch_size = 4
|
|
|
|
| 11 |
# ============================
|
| 12 |
# Configuration Parameters
|
| 13 |
# ============================
|
| 14 |
+
dataset_path = 'flpelerin/tinystories-100k'
|
| 15 |
|
| 16 |
num_epochs = 1
|
| 17 |
batch_size = 4
|