From 81d97438365490773ae2f8a997ce5e280648da77 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sun, 30 Oct 2016 11:41:52 +0000 Subject: [PATCH] export recurrent --- examples/char-rnn.jl | 2 +- src/layers/recurrent.jl | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/char-rnn.jl b/examples/char-rnn.jl index 7933c6eb..bef8fa11 100644 --- a/examples/char-rnn.jl +++ b/examples/char-rnn.jl @@ -10,7 +10,7 @@ train = zip(getbatches(input, alphabet), getbatches(input[2:end], alphabet)) model = Chain( Input(length(alphabet)), - Flux.Recurrent(length(alphabet), 128, length(alphabet)), + Recurrent(length(alphabet), 128, length(alphabet)), softmax) m = tf(unroll(model, 50)) diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl index 8882025b..b3c37406 100644 --- a/src/layers/recurrent.jl +++ b/src/layers/recurrent.jl @@ -1,3 +1,5 @@ +export Recurrent + @net type Recurrent Wxh; Whh; Why bh; by