From 04b026f5d7083ba23261be976e12193a2f7f2480 Mon Sep 17 00:00:00 2001 From: koenigmaximilian Date: Sat, 7 Jul 2018 18:49:05 +0000 Subject: [PATCH] Simplify recurrent example Produced better results when I tested (the Shakespeare example already contains a more complex recurrent network, let's KISS on this one.) --- examples/main/recurrent.hs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/main/recurrent.hs b/examples/main/recurrent.hs index b81d148..f83f612 100644 --- a/examples/main/recurrent.hs +++ b/examples/main/recurrent.hs @@ -31,10 +31,10 @@ import Grenade.Recurrent -- creation function know how to treat the layers. type R = Recurrent -type RecNet = RecurrentNetwork '[ R (LSTM 1 4), R (LSTM 4 1)] - '[ 'D1 1, 'D1 4, 'D1 1 ] +type RecNet = RecurrentNetwork '[ R (LSTM 1 1)] + '[ 'D1 1, 'D1 1] -type RecInput = RecurrentInputs '[ R (LSTM 1 4), R (LSTM 4 1)] +type RecInput = RecurrentInputs '[ R (LSTM 1 1)] randomNet :: MonadRandom m => m RecNet randomNet = randomRecurrent