diff --git a/jaxley/channels/lif.py b/jaxley/channels/lif.py index 1b491a3a..b99f2f4c 100644 --- a/jaxley/channels/lif.py +++ b/jaxley/channels/lif.py @@ -2,6 +2,7 @@ # licensed under the Apache License Version 2.0, see from typing import Dict, Optional +from warnings import warn import jax.numpy as jnp @@ -25,6 +26,8 @@ def __init__(self, name: Optional[str] = None): } self.current_name = f"{self.name}_i" + warn("This channel is not differentiable. Consider using SmoothLIF instead.") + def update_states(self, states, dt, v, params): prefix = self._name # Decrease refractory time