fix: model somato-dendritic baseline per neuron

1 files changed, 2 insertions, 0 deletions

diff --git a/sdil/probes.py b/sdil/probes.py
index 5489c6e..b58fe89 100644
--- a/sdil/probes.py
+++ b/sdil/probes.py
@@ -105,6 +105,8 @@ def _clone(net):
     n.b = [b.clone() for b in net.b]
     n.A = [a.clone() for a in net.A]
     n.P = [p.clone() for p in net.P]
+    n.P_bias = ([p.clone() for p in net.P_bias]
+                if getattr(net, "P_bias", None) is not None else None)
     n.Bnuis = [b.clone() for b in net.Bnuis]
     n.mW = [torch.zeros_like(w) for w in net.W]
     n.mb = [torch.zeros_like(b) for b in net.b]