Refactored set_seed into a standalone function and updated imports.

Author: ATATC

mipcandy/__init__.py

@@ -12,6 +12,6 @@
 from mipcandy.profiler import ProfilerFrame, Profiler
 from mipcandy.run import config
 from mipcandy.sanity_check import num_trainable_params, model_complexity_info, SanityCheckResult, sanity_check
-from mipcandy.training import TrainerToolbox, Trainer
+from mipcandy.training import TrainerToolbox, Trainer, set_seed
 from mipcandy.types import Setting, Settings, Params, Transform, SupportedPredictant, Colormap, Device, Shape2d, \
     Shape3d, Shape, AmbiguousShape, Paddings2d, Paddings3d, Paddings

mipcandy/training.py

@@ -59,6 +59,18 @@ class TrainerTracker(object):
     worst_case: int | None = None
 
 
+def set_seed(seed: int) -> None:
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    torch.backends.cudnn.benchmark = False
+    torch.backends.cudnn.deterministic = True
+    random_seed(seed)
+    np.random.seed(seed)
+    environ['PYTHONHASHSEED'] = str(seed)
+
+
 class Trainer(WithPaddingModule, WithNetwork, metaclass=ABCMeta):
     def __init__(self, trainer_folder: str | PathLike[str], dataloader: DataLoader[tuple[torch.Tensor, torch.Tensor]],
                  validation_dataloader: DataLoader[tuple[torch.Tensor, torch.Tensor]], *, recoverable: bool = True,
@@ -197,15 +209,7 @@ def set_frontend(self, frontend: type[Frontend], *, path_to_secrets: str | PathL
         self._frontend = frontend(load_secrets(path=path_to_secrets) if path_to_secrets else load_secrets())
 
     def set_seed(self, seed: int) -> None:
-        np.random.seed(seed)
-        torch.manual_seed(seed)
-        torch.cuda.manual_seed(seed)
-        torch.cuda.manual_seed_all(seed)
-        torch.backends.cudnn.benchmark = False
-        torch.backends.cudnn.deterministic = True
-        random_seed(seed)
-        np.random.seed(seed)
-        environ['PYTHONHASHSEED'] = str(seed)
+        set_seed(seed)
         if self.initialized():
             self.log(f"Set to manual seed {seed}")