zlrai5895的博客

torch.nn.utils.rnn.pack_padded_sequence()的用法： targets = pack_padded_sequence(captions, lengths, batch_first=True)[0]   输入的caption是长度不同的句子 返回的targets是打包成一条后的 captions tensor([[   1,    8, 1704,   4...

随机种子 在训练开始时，参数的初始化是随机的，为了让每次的结果一致，我们需要设置随机种子。在main函数中加入如下语句： torch.manual_seed(args.seed)#为CPU设置随机种子 if cuda:     torch.cuda.manual_seed(seed)#为当前GPU设置随机种子     torch.cuda.manual_seed_all(seed)#为所有GP...