diff --git a/src/layers/shims.jl b/src/layers/shims.jl index 8ffc4e1b..bad409b9 100644 --- a/src/layers/shims.jl +++ b/src/layers/shims.jl @@ -1,7 +1,7 @@ export Conv2D, MaxPool, Reshape type Conv2D <: Model - filter::Param{Array{Float32,4}} # [height, width, inchans, outchans] + filter::Param{Array{Float64,4}} # [height, width, inchans, outchans] stride::Dims{2} end