Minor fixes.

This commit is contained in:
Fangjun Kuang 2021-12-04 10:55:58 +08:00
parent 8df3220cb7
commit e62fe73104

View File

@ -165,7 +165,6 @@ class LayerNormLSTMCell(nn.Module):
- `next_c`: It is of shape (batch_size, hidden_size) containing the
next cell state for each element in the batch.
"""
assert check_argument_types()
if state is None:
zeros = torch.zeros(
input.size(0),