pytorch
爱生活爱自己爱学习
这个作者很懒,什么都没留下…
展开
-
pytorch加载模型torch.load() 遇到问题ModuleNotFoundError: No module named 'models'
希望完整保存模型之后,可以在另外一个地方进行测试使用,但是加载模型时遇到问题 File "/home/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 426, in load return _load(f, map_location, pickle_module, **pickle_loa...原创 2020-05-04 20:19:24 · 19052 阅读 · 9 评论 -
torch.ne()参数解释及用法
原创 2020-04-04 11:51:12 · 11241 阅读 · 2 评论 -
解决torchtext TypeError not supported between instances of Example and Example
解决torchtext TypeError '<' not supported between instances of 'Example' and 'Example'参考https://github.com/pytorch/text/issues/474原创 2020-03-16 14:18:02 · 2217 阅读 · 0 评论 -
解决错误 RuntimeError: cuda runtime error (710) : device-side assert triggered a
在github上看别人的代码,用别人的数据集跑通了,满心欢喜的换自己的数据集,修改了一番后,发现遇到了莫名其妙的错误,如下Traceback (most recent call last): File "train_discriminator.py", line 167, in <module> main() File "train_discriminator.py...原创 2020-03-13 16:36:50 · 11956 阅读 · 10 评论 -
torchtext的可复现问题
在使用torchtext.data.TabularDataset分割训练集测试集时发现遇到了奇怪的问题代码如下:self.TEXT = data.Field(init_token='<start>', eos_token='<eos>', lower=True, tokenize='spacy', fix_length=16)self.LABEL = data....原创 2020-03-13 16:26:30 · 387 阅读 · 0 评论 -
torch遇到的一些问题和知识记录
2 mask_fill_3 torch.gather 作用就是按照后面的矩阵(每一个值代表index)从前面的矩阵里取值,生成新的4原创 2019-12-26 14:32:23 · 167 阅读 · 0 评论 -
torch.repeat torch.expand
参考了https://blog.csdn.net/suian0424/article/details/99995475一个很好的例子关于repeat 请参考https://blog.csdn.net/qq_29695701/article/details/89763168原创 2019-12-24 10:23:59 · 205 阅读 · 0 评论 -
torch.squeeze()和torch.unsqueeze()的区别
torch.squeeze() 主要对数据的维度进行压缩,去掉维数为1(最前面一维,也就是最外面一维)的维度,比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.squeeze(a,N) a中去...原创 2019-12-24 09:59:24 · 197 阅读 · 0 评论 -
pytorch.arange和pytorch.range的区别
转载自https://blog.csdn.net/m0_37586991/article/details/88830026转载 2019-12-24 09:20:15 · 279 阅读 · 0 评论 -
pytorch torchtext
参考https://blog.csdn.net/qq_42564846/article/details/83584028https://blog.csdn.net/qq_42564846/article/details/83584028https://blog.csdn.net/leo_95/article/details/87708267https://blog.csdn.net/J...原创 2019-12-16 19:14:57 · 109 阅读 · 0 评论