自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(7)
  • 收藏
  • 关注

原创 CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm(...)

虽然网络训练20类,数据集10类还是可以把训练跑起来的,这样的情况就不能把训练跑起来了。所以只要去修改全连接层的最后汇集的类的数量就可以解决这个问题。我在网络上和GPT上搜索都是说什么版本错误,这个一定要注意,不要觉得版本很容易出现问题,在Debug的时候,版本是最后考虑的事情。希望可以帮助到你,我也是有点脑壳抽风所以才会出现这样的问题,如果能对看到的你Debug有帮助我也很开心。记录一下我的比较离谱的错误,如果有相同的问题可以参考一下。部分,这个部分表示了网络训练出来的类的数量是小于数据集中的数量的。

2023-09-14 10:47:30 530 1

原创 多类别分类中的Precision,Recall,F1性能计算

我在学习这三个指标的计算的时候,看了很多博客,发现很多博客对这几个性能指标的评价就好像是在说绕口令一样,让我的大脑很难理解这部分的内容,而且博客里面大都是用二分类的实验来举例如何计算这三个指标,也是不常用的,所以在这里我就用比较简单的语言来介绍这三个指标,争取让看到这篇博客的朋友们能一眼懂!首先我们这个多标签的任务中是动物分类标签,一共有12类,分别是【老鼠,牛,虎,兔,龙,蛇,马,羊,猴,坤坤,狗,猪】。注意,这三个指标的计算是按照每一类来计算的,我们在论文中看到的是多个类别的平均值。

2023-09-12 15:29:39 230

原创 网络结构分享:提取训练集特征并用于测试集检索的基本框架

在检索过程中,有直接使用训练集中提取特征用于检索,也有将训练集中的数据转化为哈希码用于检索,如果想了解如何将训练集中的数据用于哈希码的可以看我上一篇博客,在那篇博客详细讲述了如何去将训练集的数据转化为哈希码。下面介绍检索过程中使用训练集的特征来检索的方式。

2023-07-11 15:52:40 133

原创 网络结构分享:如何将网络最后的结果用哈希码的方式保存

有很多的检索相关的任务是使用哈希码的方式保存数据,然后在检索的时候将哈希码进行汉明距比较得出相似度的判断,那么如何去完成对于哈希码的保存是非常重要的一个环节。博主最近学习了如何去完成对于哈希码的保存,也是踩了很多坑。代码还在跑没事干,那么将哈希码生成步骤记录一下,分享给大家参考的同时也作为一个笔记,方便以后回顾。

2023-07-11 15:30:17 65

原创 Pytorch训练问题:torch.tensor()函数的解释及相关思考

在 PyTorch 中,torch.tensor 是一个多维矩阵,其中包含同种类型的元素(如整数或浮点数)。它是 PyTorch 库的核心类,几乎所有的计算操作都可以在 tensor 上进行。一般使用在矩阵中,那么矩阵的行列怎么在函数中体现,下面我只讨论在二维的tensor中,只有二维才会存在行列的概念。在创建中,tensor是直接可以输入数据创建,Tensor是创建---有几个维度,每个维度多大的函数。这篇博客是我对于torch框架中的tensor的一些记录和思考,会不定时更新我新的想法。

2023-07-03 10:34:06 1584

原创 Pytorch训练问题:系统找不到指定的路径

在开始写博客后又出现了相同的问题,简单记录一下,也方便有些对pytorch框架的深度学习初学者找到对应的解决方案。我在复现论文中的代码的时候,对于数据集的获取的路径有一些疑问,现在我给出对于在Windows环境下路径的一些规则,以供参考。的情况,所以在不知道1内容的情况下,一直认为是两种斜杠不能互相使用。实际情况是我的data数据保存的位置并不在这个路径的上级,而是上级的上级。我在一开始去查找问题的时候,我去本地查看路径,一直都是。在进行路径的构造方面是相同的,路径中都使用。,如果在拼接内容尾部添加。

2023-06-30 15:38:34 2128

原创 Pytorch 训练问题:indices should be either on cpu or on the same device as the indexed tensor

在这段代码中的运行中就发生了部分在GPU上运行,部分在CPU上运行的情况,我查阅了很多的相关博客,最终总结出下面通用的方法(在本博客最后将列出参考博客)。

2023-06-30 00:54:56 2305 4

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除