HuwCampbell/grenade
1
1
Fork 0
mirror of https://github.com/HuwCampbell/grenade.git synced 2024-11-22 06:55:13 +03:00
Code Issues Projects Releases Wiki Activity

Simplify recurrent example

Browse Source 
Produced better results when I tested (the Shakespeare example already contains a more complex recurrent network, let's KISS on this one.)
This commit is contained in:
koenigmaximilian 2018-07-07 18:49:05 +00:00 committed by Maximilian König
parent 93435b2b99
commit 04b026f5d7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/main/recurrent.hs
View File

@ -31,10 +31,10 @@ import Grenade.Recurrent
-- creation function know how to treat the layers. -- creation function know how to treat the layers.
type R = Recurrent type R = Recurrent
type RecNet = RecurrentNetwork '[ R (LSTM 1 4), R (LSTM 4 1)] type RecNet = RecurrentNetwork '[ R (LSTM 1 1)]
'[ 'D1 1, 'D1 4, 'D1 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 :: MonadRandom m => m RecNet
randomNet = randomRecurrent randomNet = randomRecurrent