diff --git a/src/type.jl b/src/type.jl index 28ab1b5..43b5b48 100644 --- a/src/type.jl +++ b/src/type.jl @@ -168,7 +168,7 @@ function kfn_1(params::Dict; device=cpu) # ---------------------------------------------------------------------------- # # row*col is a 2D matrix represent all RSNN activation row, col, batch = kfn.params[:inputPort][:signal][:numbers] # z-axis represent signal batch number - col += kfn.params[:inputPort][:noise][:numbers][2] + kfn.inputSize = [row, col] |> device col += kfn.params[:computeNeuron][:lif][:numbers][2] col += kfn.params[:computeNeuron][:alif][:numbers][2]