update

Author: jxtngx

examples/wandb_run.py

@@ -16,7 +16,6 @@
 from typing import Any, Dict, Optional
 
 import wandb as wb
-from lightning.pytorch import seed_everything
 from lightning.pytorch.callbacks import EarlyStopping
 from lightning.pytorch.loggers import WandbLogger
 
@@ -64,7 +63,6 @@ def __init__(
         project_name: Optional[str] = None,
         wandb_dir: Optional[str] = conf.WANDBPATH,
     ):
-        seed_everything(conf.GLOBALSEED, workers=True)
         self._wb_run = wb.init(project=project_name, dir=wandb_dir)
         self.pipeline_work = PipelineWorker(LitDataModule)
         self.training_work = TrainerWorker(

lightning_pod/core/trainer.py

@@ -31,6 +31,7 @@ def __init__(
         logger: Optional[Logger] = None,
         profiler: Optional[Profiler] = None,
         callbacks: Optional[List] = [],
+        set_seed: bool = True,
         **trainer_init_kwargs: Dict[str, Any]
     ) -> None:
         super().__init__(
@@ -40,7 +41,8 @@ def __init__(
             **trainer_init_kwargs
         )
         # SET SEED
-        seed_everything(conf.GLOBALSEED, workers=True)
+        if set_seed:
+            seed_everything(conf.GLOBALSEED, workers=True)
         #  SET DATALOADER
         self.datamodule = LitDataModule()
         #  SET MODEL