Method __init__

networks/SAR.py:145–150

(self, input_size, hidden_size)

class LSTMEncoder(nn.Module):
    def __init__(self, input_size, hidden_size):
        super(LSTMEncoder, self).__init__()
        self.input_size = input_size
        self.hidden_size = hidden_size
        self.sequence_cell = nn.LSTMCell(input_size, hidden_size)
        self.avgpool = nn.AdaptiveAvgPool2d((None, 1))

    def forward(self, features):
        batch_size = features.size(0)