Merge pull request #32 from neworderofjamie/main

Fixed missing threshold in RCuBaLIF

spyx/nn.py

@@ -354,12 +354,13 @@ def initial_state(self, batch_size):
 
 class RCuBaLIF(hk.RNNCore): 
     def __init__(self, hidden_shape, alpha=None, beta=None,  
-                 activation = superspike(),
+                 threshold = 1, activation = superspike(),
                  name="RCuBaLIF"):
         super().__init__(name=name)
         self.hidden_shape = hidden_shape
         self.alpha = alpha
         self.beta = beta
+        self.threshold = threshold
         self.spike = activation
 
     def __call__(self, x, VI):
@@ -524,4 +525,4 @@ def __init__(
 
   def __call__(self, value: jax.Array) -> jax.Array:
     return sum_pool(value, self.window_shape, self.strides,
-                    self.padding, self.channel_axis)
+                    self.padding, self.channel_axis)