
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 · 20031 阅读 · 9 评论 -
torch.ne()参数解释及用法
原创 2020-04-04 11:51:12 · 12334 阅读 · 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 · 2380 阅读 · 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 · 12232 阅读 · 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 · 444 阅读 · 0 评论 -
torch遇到的一些问题和知识记录
2 mask_fill_3 torch.gather 作用就是按照后面的矩阵(每一个值代表index)从前面的矩阵里取值,生成新的4原创 2019-12-26 14:32:23 · 210 阅读 · 0 评论 -
torch.repeat torch.expand
参考了https://blog.youkuaiyun.com/suian0424/article/details/99995475一个很好的例子关于repeat 请参考https://blog.youkuaiyun.com/qq_29695701/article/details/89763168原创 2019-12-24 10:23:59 · 265 阅读 · 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 · 253 阅读 · 0 评论 -
pytorch.arange和pytorch.range的区别
转载自https://blog.youkuaiyun.com/m0_37586991/article/details/88830026转载 2019-12-24 09:20:15 · 335 阅读 · 0 评论 -
pytorch torchtext
参考https://blog.youkuaiyun.com/qq_42564846/article/details/83584028https://blog.youkuaiyun.com/qq_42564846/article/details/83584028https://blog.youkuaiyun.com/leo_95/article/details/87708267https://blog.youkuaiyun.com/J...原创 2019-12-16 19:14:57 · 154 阅读 · 0 评论