diff options
Diffstat (limited to 'onmt/encoders/encoder.py')
-rw-r--r-- | onmt/encoders/encoder.py | 9 |
1 files changed, 0 insertions, 9 deletions
diff --git a/onmt/encoders/encoder.py b/onmt/encoders/encoder.py index 78802015..b6175638 100644 --- a/onmt/encoders/encoder.py +++ b/onmt/encoders/encoder.py @@ -2,8 +2,6 @@ import torch.nn as nn -from onmt.utils.misc import aeq - class EncoderBase(nn.Module): """ @@ -34,12 +32,6 @@ class EncoderBase(nn.Module): def from_opt(cls, opt, embeddings=None): raise NotImplementedError - def _check_args(self, src, lengths=None, hidden=None): - n_batch = src.size(1) - if lengths is not None: - n_batch_, = lengths.size() - aeq(n_batch, n_batch_) - def forward(self, src, lengths=None): """ Args: @@ -47,7 +39,6 @@ class EncoderBase(nn.Module): padded sequences of sparse indices ``(src_len, batch, nfeat)`` lengths (LongTensor): length of each sequence ``(batch,)`` - Returns: (FloatTensor, FloatTensor, FloatTensor): |