pytorch
兔子不吃胡萝卜
这个作者很懒,什么都没留下…
展开
-
pytorch之nn.Conv1d
看代码时遇到pytorch下的nn.Conv1d(),kernel_size参数不太明白,找到了这篇博客,转载链接:pytorch之nn.Conv1d详解kernel_size其实是设置卷积核的大小,卷积核的第二维的维度和词向量维度相同,另外,由于卷积默认是对最后一维卷积,所以需要将word_length这一维转换成最后一维,卷积得到的向量是[batch_size, out_size,1],后边需要跟着Relu()和max_pooling(kernel_size = max_word_length -转载 2020-10-17 10:50:53 · 560 阅读 · 0 评论 -
Pytorch 模型的保存与加载
Pytorch 模型的保存与加载想先在一部分数据跑出来几个模型,然后选择最优的加载 在另一部分数据上继续跑,加载时遇到了问题,from_pretrained()显示state_dict中的都参数没加载成功,是因为都多了一层encoder: encoder.encoder.self.attention…所以查找了pytorch保存与加载模型的相关知识,在这里记录一下~来源:https://pytorch.org/tutorials/beginner/saving_loading_models.html原创 2020-07-28 22:41:17 · 762 阅读 · 0 评论