fix

train.py

@@ -29,8 +29,13 @@ tokenizer.pad_token = tokenizer.eos_token
 vocab_size = tokenizer.vocab_size
 print(f"Tokenzier has {vocab_size} unique tokens")
 
+
+wandb.login(key="860f8753998c6e6dc356914de07e8855aa2f9642")
+
 # Initialize W&B Logging
-wandb.init(project="minGRU-Training", config={
+wandb.init(
+    project="minGRU-Training", 
+    config={
         "dataset_path": dataset_path,
         "num_epochs": num_epochs,
         "batch_size": batch_size,
@@ -42,8 +47,7 @@ wandb.init(project="minGRU-Training", config={
         "reset_state_every": reset_state_every,
         "validate_every": validate_every,
         "device": str(device)
-    },
-    settings=wandb.Settings(api_key="860f8753998c6e6dc356914de07e8855aa2f9642")
+    }
 )
 
 # ============================