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