From 5c3e38a55a3eb7f3e8442082a0ebaa6464797ca1 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 4 Jan 2023 16:08:51 +0100
Subject: [PATCH] random search initializes all random seeds
---
mu_map/training/random_search.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/mu_map/training/random_search.py b/mu_map/training/random_search.py
index 9274ddb..f2afd10 100644
--- a/mu_map/training/random_search.py
+++ b/mu_map/training/random_search.py
@@ -8,8 +8,9 @@ import shutil
import sys
from typing import Any, Callable, Dict, List, Optional
-import torch
+import numpy as np
import pandas as pd
+import torch
from mu_map.dataset.default import MuMapDataset
from mu_map.dataset.patches import MuMapPatchDataset
@@ -289,6 +290,8 @@ class RandomSearchCGAN(RandomSearch):
seed = random.randint(0, 2**32 - 1)
random.seed(seed)
+ torch.manual_seed(args.seed)
+ np.random.seed(args.seed)
self.logger.info(f"Random seed for iteration {i} is {seed}")
self._setup_run(i)
--
GitLab