diff options
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 |