fix

2023-08-18 10:06:01 +07:00
parent 1bec436408
commit d9c202ae97
3 changed files with 9 additions and 4 deletions
--- a/src/type.jl
+++ b/src/type.jl
@@ -191,7 +191,7 @@ function kfn_1(params::Dict; device=cpu)
    for slice in eachslice(w, dims=3)
        pool = shuffle!([1:row*col...])[1:synapticConnection]
        for i in pool
-            slice[i] = randn()/10   # assign weight to synaptic connection
+            slice[i] = randn()/10   # assign weight to synaptic connection, start small
        end
    end
    # project 3D w into 4D kfn.lif_wRec (row, col, n, batch)