Update train_kaggle.py

train_kaggle.py

@@ -21,6 +21,23 @@ logging.basicConfig(
 )
 
 
+def set_global_seeds(seed: int) -> None:
+    """Seed Python, NumPy, and PyTorch so that --seed controls weight init
+    as well as environment episode generation.
+    """
+    import random
+    import numpy as np
+    random.seed(seed)
+    np.random.seed(seed)
+    try:
+        import torch
+        torch.manual_seed(seed)
+        if torch.cuda.is_available():
+            torch.cuda.manual_seed_all(seed)
+    except ImportError:
+        pass
+
+
 def _add_file_logging(out_dir: Path) -> None:
     formatter = logging.Formatter("%(asctime)s | %(levelname)s | %(message)s")
     file_handler = logging.FileHandler(str(out_dir / "training.log"))
@@ -235,6 +252,7 @@ def build_model(args: argparse.Namespace, train_env):
         device=args.device,
         verbose=1,
         tensorboard_log=tb_log,
+        seed=args.seed,  # weight-init + SB3 internal RNG; env already seeded via ForecastConfig
     )
 
 
@@ -390,9 +408,11 @@ def run_training(args: argparse.Namespace) -> None:
 
 def main() -> None:
     args = _parse_args()
+    set_global_seeds(args.seed)
+    logger.info("Global seeds set to %s (Python / NumPy / PyTorch)", args.seed)
     _add_dataset_to_path(args.dataset_dir)
     run_training(args)
 
 
 if __name__ == "__main__":
-    main()
+    main()