git.iko.soy/grenade
on master help sign in
Merge pull request #70 from koenigmaximilian/patch-1
92b5760Huw Campbell
examples/main/recurrent.hs+3−3
compare with master
unifiedsplit
@@ -31,10 +31,10 @@ import Grenade.Recurrent
31-- creation function know how to treat the layers.
31-- creation function know how to treat the layers.
32type R = Recurrent
32type R = Recurrent
33
33
34−type RecNet = RecurrentNetwork '[ R (LSTM 1 4), R (LSTM 4 1)]
34+type RecNet = RecurrentNetwork '[ R (LSTM 1 1)]
35− '[ 'D1 1, 'D1 4, 'D1 1 ]
35+ '[ 'D1 1, 'D1 1]
36
36
37−type RecInput = RecurrentInputs '[ R (LSTM 1 4), R (LSTM 4 1)]
37+type RecInput = RecurrentInputs '[ R (LSTM 1 1)]
38
38
39randomNet :: MonadRandom m => m RecNet
39randomNet :: MonadRandom m => m RecNet
40randomNet = randomRecurrent
40randomNet = randomRecurrent