Update train.py

train.py

@@ -6,7 +6,9 @@ import numpy as np
 import wandb  # Import W&B library
 
 from model import minGRULM
-from util import generate_text
+from util import generate_text, generate_name
+
+
 
 # ============================
 # Configuration Parameters
@@ -101,16 +103,6 @@ print(f"Model has {parameters_count:,} parameters")
 # ============================
 # Initialize the Weights and Biases Run
 # ============================
-
-## 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",