fixing minimal_init

SciML · Oct 17, 2024 · c695740 · c695740
1 parent 5aca752
commit c695740
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/esn/esn_input_layers.jl b/src/esn/esn_input_layers.jl
@@ -190,10 +190,10 @@ function minimal_init(rng::AbstractRNG, ::Type{T}, dims::Integer...;
     return layer_matrix
 end
 
-function _create_bernoulli(p::T,
+function _create_bernoulli(p::Number,
         res_size::Int,
         in_size::Int,
-        weight::T,
+        weight::Number,
         rng::AbstractRNG,
         ::Type{T}) where {T <: Number}
     input_matrix = zeros(T, res_size, in_size)
@@ -210,7 +210,7 @@ function _create_irrational(irrational::Irrational,
         start::Int,
         res_size::Int,
         in_size::Int,
-        weight::T,
+        weight::Number,
         rng::AbstractRNG,
         ::Type{T}) where {T <: Number}
     setprecision(BigFloat, Int(ceil(log2(10) * (res_size * in_size + start + 1))))