printing ConvTranspose layer

2018-12-09 12:50:09 -05:00 · 2018-12-09 12:50:09 -05:00 · ed835f26fe
commit ed835f26fe
parent 1648414a5d
1 changed files with 3 additions and 0 deletions
--- 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}} =