pytorch
Silver_Lyan
这个作者很懒,什么都没留下…
展开
-
把tensor转成int
eg. tensor(2) -> 2tensor(2).item()https://blog.csdn.net/zl535320706/article/details/81175372原创 2020-11-02 12:41:52 · 10092 阅读 · 1 评论 -
CUDA out of memory
减小batchsize在output = model(input)前面加上 with torch.no_grad:总结:https://segmentfault.com/a/1190000022589080原创 2020-10-21 15:56:07 · 112 阅读 · 0 评论 -
AttributeError: ‘list‘ object has no attribute ‘cuda‘
AttributeError: ‘list’ object has no attribute ‘cuda’查看自己的target类型,原为[‘1’,‘0’,‘1’,‘1’]。这种列表试字符串型。而应该修改为torch.tensor类型。才能用于网络计算简单改为:先改为numpy再转换为tensor,搞定!label = torch.from_numpy(np.fromstring(label, dtype=int, sep=’,’))所以需要先把target从str映射成int,再转为tensor原创 2020-10-20 15:36:48 · 5192 阅读 · 2 评论 -
用np.loadtxt()读取txt发现数据精度甚至数据大小被改变
pytorch tensor 类型转换 (需求:int64转成tensor.inte64)决定开始记录自己做项目过程中遇到的bug啦。我利用torch.Tensor()转换int64类型的数据,转完之后tensor中元素为浮点数。一番x.dytpe()查看元素具体类型后才发现torch.Tensor()默认转成tensor.float32,如果要tensor.int64,需要LongTensor。所以没事多看官方文档很重要。...原创 2020-09-24 16:45:24 · 2524 阅读 · 0 评论