export recurrent

2016-10-30 11:41:52 +00:00 · 2016-10-30 11:41:52 +00:00 · 81d9743836
commit 81d9743836
parent fabff9b0d2
2 changed files with 3 additions and 1 deletions
--- 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))
--- a/src/layers/recurrent.jl
+++ b/src/layers/recurrent.jl
@ -1,3 +1,5 @@
+export Recurrent
+
@net type Recurrent
  Wxh; Whh; Why
  bh; by