revert this

2018-02-28 23:06:53 +00:00 · 2018-02-28 23:06:53 +00:00 · c2fea2acf6
commit c2fea2acf6
parent 2eb38eedbf
1 changed files with 2 additions and 1 deletions
--- 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