From 20e78e274ecaccc7677373a891ddadcb5777dda7 Mon Sep 17 00:00:00 2001 From: Dhairya Gandhi Date: Wed, 26 Feb 2020 22:41:45 +0530 Subject: [PATCH] docs fix --- 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 41f0e2e3..997b96e5 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -7,6 +7,7 @@ _convtransoutdims(isize, ksize, ssize, dsize, pad) = (isize .- 1).*ssize .+ 1 .+ expand(N, i::Tuple) = i expand(N, i::Integer) = ntuple(_ -> i, N) + """ Conv(filter::Tuple, in=>out) Conv(filter::Tuple, in=>out, activation) @@ -127,7 +128,7 @@ outdims(l::Conv, isize) = """ ConvTranspose(size, in=>out) - ConvTranspose(size, in=>out, relu) + ConvTranspose(size, in=>out, activation) Standard convolutional transpose layer. `filter` should be a tuple like `(2, 2)`. `in` and `out` specify the number of input and output channels respectively.