Update train.py
Browse files
train.py
CHANGED
|
@@ -14,7 +14,7 @@ from util import generate_text, generate_name
|
|
| 14 |
# Configuration Parameters
|
| 15 |
# ============================
|
| 16 |
dataset_path = 'flpelerin/tinystories-100k'
|
| 17 |
-
model_name = generate_name()
|
| 18 |
|
| 19 |
num_epochs = 1
|
| 20 |
batch_size = 4
|
|
|
|
| 14 |
# Configuration Parameters
|
| 15 |
# ============================
|
| 16 |
dataset_path = 'flpelerin/tinystories-100k'
|
| 17 |
+
model_name = generate_name() # Example: "mingru-a14c"
|
| 18 |
|
| 19 |
num_epochs = 1
|
| 20 |
batch_size = 4
|