Add init for sensitivity equations

brian-team · Mar 10, 2020 · 2c082d1 · 2c082d1
1 parent 761176f
commit 2c082d1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/brian2modelfitting/fitter.py b/brian2modelfitting/fitter.py
@@ -342,6 +342,10 @@ def setup_simulator(self, network_name, n_neurons, output_var, param_init,
 
         network = Network(neurons, monitor)
 
+        if calc_gradient:
+            param_init = dict(param_init)
+            param_init.update(get_sensitivity_init(neurons, self.parameter_names,
+                                                   param_init))
         simulator.initialize(network, param_init, name=network_name)
         return simulator