Skip to content

Commit

Permalink
Merge pull request #102 from julian121266/patch-3
Browse files Browse the repository at this point in the history
Update Clockwork_lstm layer.
  • Loading branch information
Qwlouse committed Nov 25, 2015
2 parents 3b6015b + 1dfee99 commit 2d03d14
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions brainstorm/layers/clockwork_lstm_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,16 +10,15 @@
StructureTemplate


def ClockworkLstm(size, timing, activation='tanh', name=None):
def ClockworkLstm(size, activation='tanh', name=None):
return ConstructionWrapper.create(ClockworkLstmLayerImpl,
size=size,
timing=timing,
name=name,
activation=activation)


class ClockworkLstmLayerImpl(Layer):
expected_kwargs = {'size', 'timing', 'activation'}
expected_kwargs = {'size', 'activation'}
expected_inputs = {'default': StructureTemplate('T', 'B', '...')}

computes_no_gradients_for = ['timing']
Expand Down

0 comments on commit 2d03d14

Please sign in to comment.