From ed835f26fef6a8728e77f3978d13375372990527 Mon Sep 17 00:00:00 2001 From: Tejan Karmali Date: Sun, 9 Dec 2018 12:50:09 -0500 Subject: [PATCH] printing ConvTranspose layer --- src/layers/conv.jl | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 384e9c8c..b236fc67 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -95,6 +95,9 @@ end function Base.show(io::IO, l::ConvTranspose) print(io, "ConvTranspose(", size(l.weight)[1:ndims(l.weight)-2]) + print(io, ", ", size(l.weight, ndims(l.weight)), "=>", size(l.weight, ndims(l.weight)-1)) + l.σ == identity || print(io, ", ", l.σ) + print(io, ")") end (a::ConvTranspose{<:Any,<:Any,W})(x::AbstractArray{T}) where {T <: Union{Float32,Float64}, W <: AbstractArray{T}} =