Error in Pretrained Version about arae HOT 2 OPEN

    def generate(self, hidden, maxlen, sample=True, temp=1.0):
            """Generate through decoder; no backprop"""

            batch_size = hidden.size(0)

            if self.hidden_init:
                    # initialize decoder hidden state to encoder output
                    state = (hidden.unsqueeze(0), self.init_state(batch_size))
            else:
                    state = self.init_hidden(batch_size)

            # <sos>
            self.start_symbols.data.resize_(batch_size)
            self.start_symbols.data.fill_(1)

            embedding = self.embedding_decoder(self.start_symbols)
            inputs = torch.cat([embedding, hidden], 1).unsqueeze(1)

            # unroll
            all_indices = []
            for i in range(maxlen):
                    output, state = self.decoder(inputs, state)
                    overvocab = self.linear(output.squeeze(1))

            if not sample:
                    vals, indices = torch.max(overvocab, 1)
            else:
                    # sampling
                    probs = F.softmax(overvocab/temp)
                    indices = torch.multinomial(probs, 1).squeeze(1)

            all_indices.append(indices.unsqueeze(1))

            embedding = self.embedding_decoder(indices)
            inputs = torch.cat([embedding, hidden], 1).unsqueeze(1)

            max_indices = torch.cat(all_indices, 1)

            return max_indices`