We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
lif
soft
hard
1 parent 934c676 commit 1880f8dCopy full SHA for 1880f8d
brainpy/_src/dyn/neurons/base.py
@@ -29,6 +29,7 @@ def __init__(
29
30
spk_fun: Callable = bm.surrogate.InvSquareGrad(),
31
spk_type: Any = None,
32
+ spk_reset: str = 'soft',
33
detach_spk: bool = False,
34
):
35
super().__init__(size=size,
@@ -38,6 +39,7 @@ def __init__(
38
39
sharding=sharding,
40
method=method)
41
42
+ self.spk_reset = spk_reset
43
self.spk_fun = is_callable(spk_fun)
44
self.detach_spk = detach_spk
45
self._spk_type = spk_type
0 commit comments