From ec65e2cec7daa096ec41630bdc0a5528a0d3dfce Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 22 Feb 2018 00:21:48 +0000 Subject: [PATCH] fix printing --- src/layers/recurrent.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index 41b66bda..eb227475 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -146,7 +146,7 @@ hidden(m::LSTMCell) = (m.h, m.c) treelike(LSTMCell) Base.show(io::IO, l::LSTMCell) = - print(io, "LSTMCell(", size(l.Wi, 2), ", ", size(l.Wi, 1), ")") + print(io, "LSTMCell(", size(l.Wi, 2), ", ", size(l.Wi, 1)÷4, ")") """ LSTM(in::Integer, out::Integer, σ = tanh) @@ -187,7 +187,7 @@ hidden(m::GRUCell) = m.h treelike(GRUCell) Base.show(io::IO, l::GRUCell) = - print(io, "GRUCell(", size(l.Wi, 2), ", ", size(l.Wi, 1), ")") + print(io, "GRUCell(", size(l.Wi, 2), ", ", size(l.Wi, 1)÷3, ")") """ GRU(in::Integer, out::Integer, σ = tanh)