From a1e826b888171541d32a6d59db2b14fdc62e95ff Mon Sep 17 00:00:00 2001 From: Dhairya Gandhi Date: Sun, 6 Oct 2019 05:10:56 +0530 Subject: [PATCH] fixes --- src/layers/conv.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 53d138db..c7bda4ab 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -179,7 +179,7 @@ struct DepthwiseConv{N,M,F,A,V} dilation::NTuple{N,Int} end -function DepthwiseConv(w::AbstractArray{T,N}, b::Union{Number AbstractVector{T}}, σ = identity; +function DepthwiseConv(w::AbstractArray{T,N}, b::Union{Number, AbstractVector{T}}, σ = identity; stride = 1, pad = 0, dilation = 1) where {T,N} stride = expand(Val(N-2), stride) pad = expand(Val(2*(N-2)), pad)