mirror of
https://github.com/ilyakooo0/grenade.git
synced 2024-09-11 14:45:57 +03:00
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.)
This commit is contained in:
parent
93435b2b99
commit
04b026f5d7
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user