Skip to content

Commit a81b846

Browse files
mmz33Mohammad Zeineldeen
and
Mohammad Zeineldeen
authored
LstmBlockV1 small fix (#70)
Co-authored-by: Mohammad Zeineldeen <[email protected]>
1 parent b11efc1 commit a81b846

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

i6_models/parts/lstm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def __init__(self, model_cfg: Union[LstmBlockV1Config, Dict[str, Any]]):
6161
self.cfg = LstmBlockV1Config.from_dict(model_cfg) if isinstance(model_cfg, dict) else model_cfg
6262

6363
self.dropout = self.cfg.dropout
64-
self.enforce_sorted = self.cgf.enforce_sorted
64+
self.enforce_sorted = self.cfg.enforce_sorted
6565
self.lstm_stack = nn.LSTM(
6666
input_size=self.cfg.input_dim,
6767
hidden_size=self.cfg.hidden_dim,

0 commit comments

Comments
 (0)