1.0 fixes fot conv transpose

FluxML · Sep 5, 2018 · cdaa0b4 · cdaa0b4
1 parent e138970
commit cdaa0b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/conv.jl b/src/layers/conv.jl
@@ -81,7 +81,7 @@ ConvTranspose(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity
   ConvTranspose(param(init(k..., reverse(ch)...)), param(zeros(ch[2])), σ,
        stride = stride, pad = pad, dilation = dilation)
 
-Flux.treelike(ConvTranspose)
+@treelike ConvTranspose
 
 function (c::ConvTranspose)(x)
   # TODO: breaks gpu broadcast :(