From 1e3bdcf5d1a39a7f0ea6973403abf98f6cfa3f29 Mon Sep 17 00:00:00 2001
From: manchewable <leungmanchiu123@yahoo.ca>
Date: Tue, 5 Mar 2024 22:09:17 -0800
Subject: [PATCH] check if seed is none

---
 src/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/train.py b/src/train.py
index 4adbcf442..dd71eabdd 100644
--- a/src/train.py
+++ b/src/train.py
@@ -51,7 +51,7 @@ def train(cfg: DictConfig) -> Tuple[Dict[str, Any], Dict[str, Any]]:
     :return: A tuple with metrics and dict with all instantiated objects.
     """
     # set seed for random number generators in pytorch, numpy and python.random
-    if cfg.get("seed"):
+    if cfg.get("seed") is not None:
         L.seed_everything(cfg.seed, workers=True)
 
     log.info(f"Instantiating datamodule <{cfg.data._target_}>")