![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
榴莲味的电池
这个作者很懒,什么都没留下…
展开
-
【Centernet复现】AttributeError:Can‘t pickle local object ‘get_dataset.<locals>.Dataset
一、错误描述AttributeError:Can't pickle local object 'get_dataset.<locals>.Dataset二、解决办法把涉及到numworks的值都改为0。opts.py文件里的参数num_workers默认值改为0。main.py文件里num_workers的值改为0。 val_loader = torch.utils.data.DataLoader( Dataset(opt, 'val'), batch_原创 2021-04-28 17:02:15 · 2077 阅读 · 5 评论 -
【已解决】关于Centernet复现,编译最新DCNv2出现的问题 LINK : fatal error LNK1181: 无法打开输入文件
错误描述:error: command 'D:\\vs2017\\VC\\Tools\\MSVC\\14.16.27023\\bin\\HostX86\\x64\\link.exe' failed with exit status 1181执行命令:python setup.py build develop最新DCNv2下载地址:(支持pytorch1.8)[https://codechina.csdn.net/mirrors/jinfagang/DCNv2_latest](https://c原创 2021-04-26 12:58:57 · 4689 阅读 · 9 评论 -
【深度学习】Pytorch之十万个为什么 1
一、搭建深度学习环境好多人在搭建深度学习环境,安装pytorch时,都会遇到这样一个问题。执行官网给的命令conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch时,会遇到网速慢,下载不下来。于是便出现了两种解决办法:换源,在终端Terminal上虽然执行这个命令,但不从官网那里在线下载,去镜像网站下载。另一种解决办法则是离线下载,关于离线下载,有博主给出这样的链接https://download.pytorc原创 2021-04-16 21:42:44 · 78 阅读 · 0 评论 -
【pytorch函数笔记(四)】torch.nn.MSELoss()
import torch.nn as nnnn.MSELoss(reduction = 'none')一、torch.nn.MSELoss()介绍 torch.nn.MSELoss()是一种均方误差损失函数。其公式如下:ln=(xn−yn)2l_n={(x_n-y_n)}^2 ln=(xn−yn)2 其中,xnx_nxn表示预测值矩阵,yny_nyn表示真实值矩阵。二、torch.nn.MSELoss()应用代码:import torchimport torch.nn as原创 2021-03-31 19:22:20 · 1932 阅读 · 0 评论 -
【pytorch函数笔记(三)】torch.nn.BCELoss()
import torch.nn as nnnn.BCELoss((weight=None, size_average=None, reduce=None, reduction=‘mean’))一、torch.nn.BCELoss()介绍 BCELoss()是计算目标值和预测值之间的二进制交叉熵损失函数。其公式如下:ln=−wn⋅[yn⋅logxn+(1−yn)⋅log(1−xn)] l_n=-w_n·[{y_n·logx_n}+{(1-y_n)·log(1-x_n)}] ln=−wn⋅[yn原创 2021-03-31 19:06:57 · 2693 阅读 · 0 评论 -
【pytorch函数笔记(二)】torch.nn.Sigmoid()
import torch.nn as nntorch.nn.sigmoid()一、sigmoid介绍 sigmoid是激活函数的一种,它会将样本值映射到0到1之间。 sigmoid的公式如下:11+e−x \frac{1}{1+e^{-x}} 1+e−x1二、sigmoid的应用代码:import torch.nn as nnimport torch#取一组满足标准正态分布的随机数构成3*3的张量t1 = torch.randn(3,3)m = nn.Sigmoid()原创 2021-03-31 18:49:05 · 45983 阅读 · 3 评论 -
【pytorch函数笔记(一)】torch.squeeze(input)
import torchtorch.squeeze(input)一、squeeze的作用是什么?压缩张量维度,即当张量的某个维度的成员数量为1时,直接删除该维度。(注意,张量的维度由外层向内层看,分别是0维,1维,2维···)二、对squeeze的作用的实例展示三、应用代码一:#以一个三维张量为例t1 = torch.tensor([[[0,1,2],[3,4,5]]]) #torch.size([1,2,3])t2 = torch.tensor([[[0],[1]]])原创 2021-03-31 15:58:17 · 262 阅读 · 0 评论 -
【Pytorch的坑】OSError: [WinError 126] 找不到指定的模块caffe2_detectron_ops_gpu.dll” or one of its dependencies
错误描述:OSError: [WinError 126] 找不到指定的模块。Error loading “D:\Anaconda3\envs\MyCode\Lib\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll” or one of its dependencies思路描述:查阅了csdn很多回答,也安装了Microsoft Visual C++ Redistributable,依旧未能解决。这个error提示的是找不到caffe2.原创 2021-03-24 11:39:24 · 31298 阅读 · 20 评论