pytorch
月半rai
这个作者很懒,什么都没留下…
展开
-
pytorch学习(一):torch.nn.utils.rnn.pack_padded_sequence()的用法
torch.nn.utils.rnn.pack_padded_sequence()的用法:targets = pack_padded_sequence(captions, lengths, batch_first=True)[0] 输入的caption是长度不同的句子 返回的targets是打包成一条后的captionstensor([[ 1, 8, 1704, 4...原创 2018-10-12 20:51:35 · 4900 阅读 · 0 评论 -
pytorch学习(二):CPU和GPU随机种子的选择
随机种子在训练开始时,参数的初始化是随机的,为了让每次的结果一致,我们需要设置随机种子。在main函数中加入如下语句:torch.manual_seed(args.seed)#为CPU设置随机种子if cuda: torch.cuda.manual_seed(seed)#为当前GPU设置随机种子 torch.cuda.manual_seed_all(seed)#为所有GP...原创 2018-12-10 22:00:39 · 3545 阅读 · 2 评论