diff --git a/ngclearn/components/neurons/spiking/LIFCell.py b/ngclearn/components/neurons/spiking/LIFCell.py index d8a0d763..af6c6683 100644 --- 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']