Skip to content

Commit a570838

Browse files
committed
Lstm, remove rec_weight_dropout option
We will handle that in a more generic way (#59).
1 parent acfad48 commit a570838

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

nn/rec.py

+1-10
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,5 @@ class Lstm(_Rec):
99
"""
1010
LSTM
1111
"""
12-
def __init__(self, n_out: int, *, rec_weight_dropout=0, rec_weight_dropout_shape=None, **kwargs):
13-
assert "unit_opts" not in kwargs, "we handle that here"
14-
unit_opts = {}
15-
if rec_weight_dropout:
16-
unit_opts["rec_weight_dropout"] = rec_weight_dropout
17-
if rec_weight_dropout_shape:
18-
unit_opts["rec_weight_dropout_shape"] = rec_weight_dropout_shape
19-
if unit_opts:
20-
kwargs = kwargs.copy()
21-
kwargs["unit_opts"] = unit_opts
12+
def __init__(self, n_out: int, **kwargs):
2213
super().__init__(n_out=n_out, unit="nativelstm2", **kwargs)

0 commit comments

Comments
 (0)