fixed minor load bug in lif-cell

NACLab · Jul 27, 2024 · 03940e9 · 03940e9
1 parent 9c932b1
commit 03940e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ngclearn/components/neurons/spiking/LIFCell.py b/ngclearn/components/neurons/spiking/LIFCell.py
@@ -307,7 +307,7 @@ def save(self, directory, **kwargs):
     def load(self, directory, seeded=False, **kwargs):
         file_name = directory + "/" + self.name + ".npz"
         data = jnp.load(file_name)
-        self.thr_theta.set(data['thr_theta'])
+        self.thr_theta.set(data['threshold_theta'])
         ## constants loaded in
         self.tau_m = data['tau_m']
         self.thr = data['thr']