From 214f71f49273e135cfcffb93d87b92e041139128 Mon Sep 17 00:00:00 2001 From: Dhairya Gandhi Date: Sun, 6 Oct 2019 04:55:33 +0530 Subject: [PATCH] add N --- src/layers/conv.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index f02adb41..53d138db 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -50,7 +50,8 @@ function Conv(w::AbstractArray{T,N}, b::Union{Number, AbstractVector{T}}, σ = i return Conv(σ, w, b, stride, pad, dilation) end -convweight(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}; init = glorot_uniform) = init(k..., ch...) +convweight(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}; + init = glorot_uniform) where N = init(k..., ch...) const convbias = zeros function Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity;