Skip to content

Commit

Permalink
fixup! lstm: Add LSTMMechanism + compiled support
Browse files Browse the repository at this point in the history
  • Loading branch information
SamKG committed Jul 25, 2020
1 parent 2a8a22b commit 5a083e6
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def __init__(self,

if default_variable is not None:
size = default_variable.shape
assert(len(default_variable)) == 3, "Must specify default variables for input, hidden state, and cell state!"
assert len(default_variable) == 3, "Must specify default variables for input, hidden state, and cell state!"
else:
assert size is not None and len(size) == 2, "Must only specify input and hidden sizes!"
size = (*size, size[1])
Expand Down

0 comments on commit 5a083e6

Please sign in to comment.