diff options
author | Alex Auvolat <alex@adnab.me> | 2017-01-17 21:58:13 +0100 |
---|---|---|
committer | Alex Auvolat <alex@adnab.me> | 2017-01-17 21:58:13 +0100 |
commit | 9eafa549291f707837940243b7302ce552eaaa20 (patch) | |
tree | 51823241f10d9b399ef139ab9ac090d3e3be4af3 /model | |
parent | 2a4f89c41b82e7173d6fb4481af0dcd58c812c19 (diff) | |
download | text-rnn-9eafa549291f707837940243b7302ce552eaaa20.tar.gz text-rnn-9eafa549291f707837940243b7302ce552eaaa20.zip |
Change hierarchy.
Diffstat (limited to 'model')
-rw-r--r-- | model/condlstm.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/model/condlstm.py b/model/condlstm.py index 1d62b0a..26d9582 100644 --- a/model/condlstm.py +++ b/model/condlstm.py @@ -10,7 +10,7 @@ from blocks.roles import WEIGHT from blocks.graph import ComputationGraph, apply_noise, apply_dropout -class CondLSTM(LSTM): +class RstLSTM(LSTM): @recurrent(sequences=['inputs', 'run_mask', 'rst_in_mask', 'rst_out_mask'], states=['states', 'cells'], contexts=[], outputs=['states', 'cells', 'outputs']) @@ -24,7 +24,6 @@ class CondLSTM(LSTM): states, cells = self.apply(iterate=False, inputs=inputs, states=states, cells=cells, mask=run_mask) - outputs = states if rst_out_mask: @@ -69,7 +68,7 @@ class Model(): bricks.append(linear) inter = linear.apply(recvalues[:, :, :indim]) - lstm = CondLSTM(dim=p['dim'], activation=config.activation_function, + lstm = RstLSTM(dim=p['dim'], activation=config.activation_function, name="lstm_rec_%d"%i) bricks.append(lstm) |