From c2fea2acf60cc33301495020cbf545e320525add Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 28 Feb 2018 23:06:53 +0000 Subject: [PATCH] revert this --- 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 84bef9f6..994648c2 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -33,7 +33,8 @@ Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = Flux.treelike(Conv) function (c::Conv)(x) - ndims(x) == ndims(c.weight)-1 && return squeezebatch(c(reshape(x, size(x)..., 1))) + # TODO: breaks gpu broadcast :( + # ndims(x) == ndims(c.weight)-1 && return squeezebatch(c(reshape(x, size(x)..., 1))) σ, b = c.σ, reshape(c.bias, map(_->1, c.stride)..., :, 1) σ.(conv(x, c.weight, stride = c.stride, pad = c.pad) .+ b) end