From dbecfb39742c1d01f39bfaec4f83b0f263753659 Mon Sep 17 00:00:00 2001
From: jasperalbers <jasperalbers@posteo.de>
Date: Wed, 19 Jan 2022 13:17:33 +0100
Subject: [PATCH] update use of rng seed
---
multiarea_model/default_params.py | 2 +-
multiarea_model/simulation.py | 19 +++++++++++--------
2 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/multiarea_model/default_params.py b/multiarea_model/default_params.py
index 499bfae..c34b2da 100644
--- a/multiarea_model/default_params.py
+++ b/multiarea_model/default_params.py
@@ -38,7 +38,7 @@ Simulation parameters
"""
sim_params = {
# master seed for random number generators
- 'master_seed': 0,
+ 'rng_seed': 0,
# simulation step (in ms)
'dt': 0.1,
# simulated time (in ms)
diff --git a/multiarea_model/simulation.py b/multiarea_model/simulation.py
index 610c317..a3f1e68 100644
--- a/multiarea_model/simulation.py
+++ b/multiarea_model/simulation.py
@@ -145,7 +145,7 @@ class Simulation:
Prepare NEST Kernel.
"""
nest.ResetKernel()
- master_seed = self.params['master_seed']
+ rng_seed = self.params['rng_seed']
num_processes = self.params['num_processes']
local_num_threads = self.params['local_num_threads']
vp = num_processes * local_num_threads
@@ -153,15 +153,18 @@ class Simulation:
'total_num_virtual_procs': vp,
'overwrite_files': True,
'data_path': os.path.join(self.data_dir, 'recordings'),
- 'print_time': False,
- 'grng_seed': master_seed,
- 'rng_seeds': list(range(master_seed + 1,
- master_seed + vp + 1))})
-
+ 'print_time': False})
+ if self.network.params['USING_NEST_3']:
+ nest.SetKernelStatus({'rng_seed': rng_seed})
+ else:
+ nest.SetKernelStatus({'grng_seed': rng_seed,
+ 'rng_seeds': list(range(rng_seed + 1,
+ rng_seed + vp + 1))})
+ self.pyrngs = [np.random.RandomState(s) for s in list(range(
+ rng_seed + vp + 1, rng_seed + 2 * (vp + 1)))]
nest.SetDefaults(self.network.params['neuron_params']['neuron_model'],
self.network.params['neuron_params']['single_neuron_dict'])
- self.pyrngs = [np.random.RandomState(s) for s in list(range(
- master_seed + vp + 1, master_seed + 2 * (vp + 1)))]
+
def create_recording_devices(self):
"""
--
GitLab