Made lambda and alpha of eltype(x)

2019-03-07 23:42:38 +05:30 · 2019-03-07 23:42:38 +05:30 · c6e51f5cc2
commit c6e51f5cc2
parent 47c1324476
1 changed files with 2 additions and 2 deletions
--- a/src/layers/normalise.jl
+++ b/src/layers/normalise.jl
@ -61,8 +61,8 @@ end

 function (a::AlphaDropout)(x)
  a.active || return x
-  λ = 1.0507009873554804934193349852946
-  α = 1.6732632423543772848170429916717
+  λ = eltype(x)(1.0507009873554804934193349852946)
+  α = eltype(x)(1.6732632423543772848170429916717)
  α1 = eltype(x)(-λ*α)
  noise = randn(eltype(x), size(x))
  x = @. x*(noise > (1 - a.p)) + α1 * (noise <= (1 - a.p))