7982a664d710a604a5157c542180a20c87cf4216,onmt/modules/ImageEncoder.py,ImageEncoder,forward,#ImageEncoder#Any#,39

Before Change

        hidden_t = []
        for row in range(input.size(2)):
            print(input[:, :, row, :].size())
            row_outputs, row_hidden_t = self.rnn(input[:, :, row, :], None)
            hidden_t.append(row_hidden_t)
        return hidden_t

After Change

        // // (batch_size, H, W, 512)
        all_outputs = []
        for row in range(input.size(2)):
            inp = input[:, :, row, :].transpose(0, 2).transpose(1, 2)
            outputs, hidden_t = self.rnn(inp)
            all_outputs.append(outputs)
        out = torch.cat(all_outputs, 0)
        
        return hidden_t, out

Instances