From eadf96605cc611a095dd2e3d4aab6d7e0879f36c Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Mon, 20 Mar 2017 23:11:08 +0000 Subject: [PATCH] fix recurrent layer batching --- src/layers/recurrent.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index e012a215..ebe8247a 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -9,7 +9,7 @@ export Recurrent, GatedRecurrent, LSTM end Recurrent(in, out; init = initn) = - Recurrent(init((in, out)), init((out, out)), init(out), init(out)) + Recurrent(init((in, out)), init((out, out)), init(1, out), init(1, out)) @net type GatedRecurrent Wxr; Wyr; br