diff options
| author | Alex Auvolat <alex@adnab.me> | 2016-04-25 11:22:39 +0200 |
|---|---|---|
| committer | Alex Auvolat <alex@adnab.me> | 2016-04-25 11:22:39 +0200 |
| commit | 760587b5d9771257160fac216dfcfff852de3ccc (patch) | |
| tree | 5625ccd99bf3d214ac686d05691c70c9514c1a55 /config/lstm-frigo-irc.py | |
| parent | 647dce75ba7483b8fa0b64ef64ecaab70932b769 (diff) | |
| download | text-rnn-760587b5d9771257160fac216dfcfff852de3ccc.tar.gz text-rnn-760587b5d9771257160fac216dfcfff852de3ccc.zip | |
Add Xreg
Diffstat (limited to 'config/lstm-frigo-irc.py')
| -rw-r--r-- | config/lstm-frigo-irc.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/config/lstm-frigo-irc.py b/config/lstm-frigo-irc.py index 4ae21e2..a0f5b5c 100644 --- a/config/lstm-frigo-irc.py +++ b/config/lstm-frigo-irc.py @@ -12,7 +12,11 @@ num_seqs = 50 seq_len = 5000 seq_div_size = 200 -hidden_dims = [1024, 1024, 1024] +layers = [ + {'dim': 1024}, + {'dim': 1024}, + {'dim': 1024}, +] activation_function = Tanh() i2h_all = True # input to all hidden layers or only first layer |