a0dadd2f2f5a2f443dfabdf696809d5e6ca8e2b5,ch12/libbots/model.py,,pack_batch,#Any#Any#Any#,117

Before Change

    output_seq_list = []
    for out in output_idx:
        seq = out[:-1]
        out_v = Variable(torch.LongTensor([seq]))
        if cuda:
            out_v = out_v.cuda()
        emb_out_v = embeddings(out_v)
        out_seq = rnn_utils.pack_padded_sequence(emb_out_v, [len(seq)], batch_first=True)
        output_seq_list.append(out_seq)
    return emb_input_seq, output_seq_list, input_idx, output_idx

After Change

    // prepare output sequences, with end token stripped
    output_seq_list = []
    for out in output_idx:
        output_seq_list.append(pack_input(out[:-1], embeddings, cuda))
    return emb_input_seq, output_seq_list, input_idx, output_idx

Instances