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

Merge pull request #70 from koenigmaximilian/patch-1

Browse Source 
Simplify recurrent example
This commit is contained in:
Huw Campbell 2018-07-11 15:29:04 +10:00 committed by GitHub
commit 92b5760926
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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.
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