torch
zhe_csdn
不断犯错不断改错不断进步的过程
展开
-
torch与tf中的normalization
torch与tf中的normalization。原创 2022-11-23 15:27:47 · 311 阅读 · 0 评论 -
检测模型改进—OHEM与Focal-Loss算法总结
OHEM,Focal-loss讲解的很详细转载 2022-08-19 16:35:39 · 209 阅读 · 0 评论 -
FCN的学习及理解(作者写得很详细,收藏一波)
https://blog.csdn.net/qq_36269513/article/details/80420363转载 2020-11-22 16:48:43 · 363 阅读 · 0 评论 -
AttributeError: 'torch._C.Value' object has no attribute 'debugName'
我的环境:tensorboardX1.9解决办法降版本tensorboardX1.8原创 2019-11-26 19:12:44 · 1283 阅读 · 0 评论 -
IndexError: dimension specified as 0 but tensor has no dimensions
loss = loss.unsqueeze(0) 升维losses.update(loss.item(), loss.size(0))IndexError: dimension specified as 0 but tensor has no dimensionsa = torch.tensor([1])b = torch.tensor(1)a 和 b 之间是有很大的区别的...原创 2019-09-20 15:12:00 · 3008 阅读 · 1 评论 -
torch的模型保存和加载 各种细节各种坑 尤其是多GPU训练会出现各种问题
#保存和加载整个模型torch.save(model_object, ‘model.pth’)model = torch.load(‘model.pth’)#仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), ‘params.pth’)model_object.load_state_dict(torch.load(‘params.p...原创 2019-09-18 11:37:05 · 5443 阅读 · 0 评论 -
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
pytorch报错:IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python一般是torch版本的不同造成的。解决:将代码中的loss_return.data[0] 改成loss_return.datanorm += params.norm(2)....转载 2019-09-17 11:13:10 · 148 阅读 · 0 评论 -
PIL opencv 与tensor之间的相互转换
#--coding:utf-8--import torchfrom torchvision import transformsfrom PIL import Imageimport cv2img_path = “./cat.59.jpg”transform1 = transforms.Compose([transforms.CenterCrop((224,224)), # 只能对PI...原创 2019-09-16 09:09:26 · 4282 阅读 · 1 评论 -
pytorch contiguous
contiguous:view只能用在contiguous的variable上。如果在view之前用了transpose, permute等,需要用contiguous()来返回一个contiguous copy。一种可能的解释是:有些tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,...原创 2019-09-10 19:44:08 · 188 阅读 · 0 评论 -
dl框架支持的算子(欢迎补充更正)
支持的算子● CaffeAccuracy, BatchNorm, Clip, Concat, Convolution, Data, Deconvolution, DepthwiseConvolution, DetectionOutput, Dropout, Eltwise, Flatten, InnerProduct, Input, LRN, Normalize, PReLU, Permute...原创 2019-09-09 16:25:16 · 560 阅读 · 0 评论 -
Apex混合精度加速 Pytorch
https://chenyue.top/2019/05/21/%E5%B7%A5%E7%A8%8B-%E4%BA%94-apex%E6%B7%B7%E5%90%88%E7%B2%BE%E5%BA%A6%E5%8A%A0%E9%80%9F/github 开源地址:https://github.com/NVIDIA/apex ,文档地址是:https://nvidia.github.io/apex...转载 2019-08-29 15:33:56 · 1221 阅读 · 0 评论 -
conda/pip安装pytorch各种版本的超详细的教程
使用conda/pip安装pytorch 0.3.1教程(mac/windos/linux)pytorch v0.3.1发布,添加余弦退火方案来调整学习率以及添加random_split/reduce等函数,其实更新有一段时间了,很多小伙伴安装的时候发现被墙了无法使用,所以我整理了一下方便大家使用。你可以参考自己的服务器和python版本进行选择:一、使用conda安装pytorch 0.3....原创 2019-06-01 15:45:32 · 24354 阅读 · 4 评论 -
pytorch中的细节问题
1 nn.Module.cuda() 和 Tensor.cuda() 的作用效果差异:对于nn.Module:model = model.cuda()model.cuda()上述两句能够达到同样的效果,即对Model自身进行内存迁移2 对于Tensor:nn.Module不同,调用tensor.cuda()只是返回这个tensor对象在GPU内存上的拷贝,而不会对自身进行改变因此必须...原创 2019-07-09 21:20:12 · 1770 阅读 · 0 评论 -
odd_errors(持续更新,欢迎补充,工程中遇到的错误)
TypeError: ‘module’ object is not subscriptableRuntimeError: merge_sort: failed to synchronize: an illegal memory access was encounteredRuntimeError: CUDA out of memory. Tried to allocate 322.00 Mi...原创 2019-06-26 13:42:44 · 1379 阅读 · 0 评论 -
pytorch的Tensor变量之间的转换
系统默认是torch.FloatTensor类型data = torch.Tensor(2,3)是一个2*3的张量,类型为FloatTensordata.cuda()就转换为GPU的张量类型,torch.cuda.FloatTensor类型(1)CPU或GPU之间的张量转换在Tensor后加long(), int(), double(),float(),byte()等函数就能将Tensor...原创 2019-06-25 20:52:35 · 19201 阅读 · 0 评论 -
pytorch odd erros
TypeError: ‘module’ object is not subscriptable文件传入失败RuntimeError: merge_sort: failed to synchronize: an illegal memory access was encounteredpytorch版本问题 0.3.1原创 2019-06-03 16:17:33 · 342 阅读 · 1 评论 -
conda命令更改pytorch版本
降级到以前的v0.1.10版本 conda install pytorch=0.1.10 -c soumith可以参考官网的配置选中你的配置,比如python版本,CUDA版本等等,然后复制代码,将版本号改为你想要的版本,如torch-0.4.1改为torch-0.1.10,不过有些配置可能没有,因为pytorch提供的支持越来越多。如果你当前有安装最新的pytorch,参考卸载之:Pyto...原创 2019-03-27 20:56:15 · 15388 阅读 · 0 评论 -
pytorch各个版本的下载地址
pytorch0.4.0, python3.5, Linux, cu80, 86_64http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whlpytorch0.4.0, python3.5, Linux, cu90, 86_64http://download.pytorch.org/whl...转载 2019-03-27 20:16:17 · 4186 阅读 · 5 评论