refractoring

2023-05-17 14:28:43 +07:00
parent df26a01929
commit 214466d9e9
5 changed files with 36 additions and 53 deletions
--- a/src/snn_utils.jl
+++ b/src/snn_utils.jl
@@ -19,7 +19,7 @@ function timestep_forward!(x::passthrough_neuron)
    x.z_t = x.z_t1
 end

-function timestep_forward!(x::compute_neuron)
+function timestep_forward!(x::Union{compute_neuron, output_neuron})
    x.z_t = x.z_t1
    x.v_t = x.v_t1
 end