Update train.py

train.py

@@ -12,6 +12,7 @@ from util import generate_text
 # Configuration Parameters
 # ============================
 dataset_path  = 'flpelerin/tinystories-100k'
+run_name      = generate_name()
 
 num_epochs    = 1
 batch_size    = 4
@@ -102,9 +103,18 @@ print(f"Model has {parameters_count:,} parameters")
 # ============================
 
 ## TODO: Add naming of the run, from: "[MODEL_NAME]-[RANDOM_HEX]" (like: mingru-a14b)
+
+def generate_name():
+    prefix = "mingru"
+    random_number = random.randint(0, 0xFFFF)
+    hex_code = f"{random_number:04x}"
+    unique_name = f"{prefix}-{hex_code}"
+    return unique_name
+
 wandb.login(key="860f8753998c6e6dc356914de07e8855aa2f9642")
 wandb.init(
-    project="minGRU-Training", 
+    project="minGRU-Training",
+    name=run_name,
     config={
         "dataset_path": dataset_path,