flpelerin commited on
Commit
9b48427
·
verified ·
1 Parent(s): a563d57

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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()} # Example: "mingru-a14c"
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