pytorch
chen645096127
这个作者很懒,什么都没留下…
展开
-
安装 detectron2 编译detectron 报错:running setup.py develop for detectron2 error: command errored out with
如下:Running setup.py install for detectron2 ... error ERROR: Command errored out with exit status 1: command: /home/jackytsai/anaconda3/envs/detectron2/bin/python -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-...原创 2020-12-22 18:39:08 · 4039 阅读 · 7 评论 -
PyTorch中 tensor.detach() 和 tensor.data 的区别
PyTorch0.4中,.data 仍保留,但建议使用 .detach(), 区别在于 .data 返回和 x 的相同数据 tensor, 但不会加入到x的计算历史里,且require s_grad = False, 这样有些时候是不安全的, 因为 x.data 不能被 autograd 追踪求微分 。 .detach() 返回相同数据的 tensor ,且 requires_grad=False...转载 2019-12-05 17:24:15 · 705 阅读 · 0 评论 -
Pytorch:多GPU训练网络与单GPU训练网络保存模型的区别
分类专栏:PyTorch测试环境:Python3.6 + Pytorch1.1在pytorch中,使用多GPU训练网络需要用到 【nn.DataParallel】: gpu_ids = [0, 1, 2, 3] device = t.device("cuda:0" if t.cuda.is_available() else "cpu") # 只能单GPU运行...转载 2019-11-21 14:07:52 · 385 阅读 · 0 评论 -
Python:IOError: image file is truncated 的解决办法
在运行demo是出现标题上述异常,解决办法是,再添加如下2句代码:from PIL import ImageFileImageFile.LOAD_TRUNCATED_IMAGES = True原创 2019-09-25 16:50:15 · 1011 阅读 · 0 评论 -
pyorch 训练分类网络的流程
1.准备、训练和测试自己的图片数据这里使用的是fashion-mnist。这个数据比较小,关于fashion-mnist数据,可以百度,也可以点此了解一下,数据如下图所示:下载地址:https://github.com/zalandoresearch/fashion-mnist训练流程主要分为以下几个步骤:加载数据,训练模型,测试模型,demo别如下:(1)加载数据:...原创 2019-08-28 19:47:57 · 479 阅读 · 0 评论 -
imgaug数据增强库——学习笔记
imgaug是一个封装好的用来进行图像augmentation的python库.库里面的一些数据增强函数在下方链接里面做了详细的说明:https://blog.csdn.net/qq_38451119/article/details/82428612官方文档链接如下:https://github.com/aleju/imgaug...原创 2019-07-26 17:12:01 · 314 阅读 · 0 评论 -
Pytorch:transforms的二十二个方法
文章目录一、 裁剪——Crop 1.随机裁剪:transforms.RandomCrop 2.中心裁剪:transforms.CenterCrop 3.随机长宽比裁剪 transforms.RandomResizedCrop 4.上下左右中心裁剪:transforms.FiveCrop 5.上下左右中心裁剪后翻转: transforms.TenCrop 二、翻转和旋转...转载 2019-07-26 16:55:13 · 876 阅读 · 0 评论 -
pytorch之expand,gather,squeeze,sum,contiguous,softmax,max,argmax
目录gathersqueezeexpandsumcontiguoussoftmaxmaxargmaxgathertorch.gather(input,dim,index,out=None)。对指定维进行索引。比如4*3的张量,对dim=1进行索引,那么index的取值范围就是0~2.input是一个张量,index是索引张量。input和index的si...转载 2019-07-29 14:37:23 · 188 阅读 · 0 评论 -
解决 THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument
如题,原因是显卡用的RTX 2080Ti,CUDA就要装10以上,这个时候pytorch不能直接用pip装,要这样:pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl源码装应该也可以。...转载 2019-07-06 09:05:06 · 11455 阅读 · 2 评论 -
module 'torchvision.transforms' has no attribute 'Resize'
torchvision版本较低pip install --upgrade torchvision 升级torchvision到 0.2.1 版本。问题解决。如果网速不好,可以再更新命令语句中添加清华镜像安装.原创 2019-07-06 08:58:12 · 5263 阅读 · 1 评论 -
【error】RuntimeError: size mismatch与全连接fc层
今天跑代码的时候遇到了这个错误:RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 128] ,at /opt/conda/conda-bld/pytorch_1524原因分析:这里是最后一层卷积层铺平后(conv_out.view(conv_out.size(0), -1))的shape与全连接层的输入不匹配导致的,因此要将维度对齐....原创 2019-06-28 17:15:41 · 5351 阅读 · 0 评论 -
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
报错原因分析:train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.解决方法:#将原语句:train_loss+=loss.data[0]#修改为:train_loss+=loss.item()#bingo...原创 2019-06-28 17:01:20 · 40118 阅读 · 22 评论 -
ImportError: cannot import name '_validate_lengths'
找到:Anaconda3/lib/python3.6/site-packages/numpy/lib/arraypad.py 954行,添加下面两个函数保存,重新加载即可消除错误 def _normalize_shape(ndarray, shape, cast_to_int=True): """ Private function which does s...转载 2019-07-02 08:54:57 · 99 阅读 · 0 评论 -
pip 临时使用清华大学镜像安装插件
当你网速不好时,在线安装安装包时,可以选择清华镜像,安装速度比之前快了很多倍,安装命令语句很简单,如下:pip install -ihttps://pypi.tuna.tsinghua.edu.cn/simplesome-package注意,simple 不能少, 是 https 而不是 http...原创 2019-07-02 08:52:14 · 1069 阅读 · 0 评论