Update config.py

config.py

@@ -2,9 +2,9 @@ from pathlib import Path
 
 def get_config():
     return {
-        "batch_size": 16, 
+        "batch_size": 8, 
         "num_epochs": 40,  
-        "lr": 5e-4, 
+        "lr": 10**-4, 
         "seq_len": 512,
         "d_model": 512, 
         "datasource": "opus_books",
@@ -12,7 +12,7 @@ def get_config():
         "lang_tgt": "fr",
         "model_folder": "weights_fr",
         "model_basename": "fr_model_",
-        "preload": None,  # Start fresh
+        "preload": None, 
         "tokenizer_file": "tokenizer_{0}.json",
         "experiment_name": "runs/oratio_en_fr"
     }