Pytorch踩坑之路（随时补充）

于 2024-02-16 14:51:13 发布
阅读量835 收藏 10
点赞数 4
分类专栏： bert 情感分析 pytorch 文章标签： pytorch 人工智能 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/2401_82949788/article/details/136128350
版权
文章讲述了在使用PyTorch时遇到的三个常见问题：数据集划分后index未重置引发KeyError，标签不匹配和类型不对应导致的RuntimeError，以及如何通过重置index、调整标签和转换数据类型来解决这些问题。
摘要由CSDN通过智能技术生成

1、Pytorch: “KeyError: Caught KeyError in DataLoader worker process 0.“

原因：df划分数据集之后没有重置index

解决办法：重置index

df_train.reset_index(drop=True, inplace=True)

参考链接：

machine learning - PyTorch: "KeyError: Caught KeyError in DataLoader worker process 0." - Stack Overflow

2、RuntimeError:CUDA error:

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

最低0.47元/天 解锁文章
山楂橙子
关注
专栏目录
pytorch踩坑记
m0_46545833的博客
03-21 1841
pytorch神经网络
【KeyError：Caught KeyError in Dataloader worker process 0. KeyError：‘标签’】
m0_61878076的博客
09-16 9505
用RefineDet训练单类别数据集时出现了两个错误。第一个KeyError是数据集加载相关的；第二个KeyError是使用了不存在的字典键值，即没读到图片标注的标签（scratch）。出现类似问题，建议先直接解决第二个，因为第一个很多时候是附带出错。
Pytorch: “KeyError: Caught KeyError in DataLoader worker process 0.“
weixin_51398304的博客
12-23 7287
转载：划分训练集的之后，没有重置索引。machine learning - Pytorch: "KeyError: Caught KeyError in DataLoader worker process 0." - Stack Overflow
Caught KeyError in DataLoader worker process 0.
qq_45160840的博客
11-28 1896
做目标检测网络训练时，遇到下面报错：KeyError：Sa**（自己数据集的一种类别）
迭代DataLoader时出现TypeError: Caught TypeError in DataLoader worker process 0.TypeError: 'NoneType' obj。
清凌的博客
02-20 4万+
解决方案： 一般遇到这样的错误，都是因为在python里面自定义的class类别读取出错。这里的 Caught TypeError in DataLoader worker process 0.意思是多线程执行数据读取任务时，遇到了类型错误， 一般还会抛出异常如下： TypeError: 'NoneType' object is not subscriptable 说明可能没有读取到数据，所以返回...
openpcdet 训练数据集报错：KeyError: ‘road_plane‘
qq_44718979的博客
04-18 1145
OpenPCDet训练报错
PyTorch踩坑指南（1）nn.BatchNorm2d()函数
白水煮蝎子
04-16 2万+
前言 最近在研究深度学习中图像数据处理的细节，基于的平台是PyTorch。心血来潮，总结一下，好记性不如烂笔头。 Batch Normalization 对于2015年出现的Batch Normalization1，2018年的文章Group Normalization2在Abstract中总结得言简意赅，我直接copy过来。 Batch Normalization (BN) is a mile...
PyTorch踩坑记录
nan1104的博客
10-04 1296
使用Pytorch也一年多了，记录一下自己遇到的各种问题。 各种踩坑记录 loss.backward()报错 一种非常常见的错误，在网络前向传播时没问题，但是当loss.backward()时会报错 导致这个错误的原因非常多样 in-place操作导致，具体的in-place操作有很多，例如squeeze_(), x[:]=y, 等等，网上相关资料很多，不细写 loss需要是一个标量，如果是向量的话，需要x.sum(); 有些向量不需要传递梯度时，使用x.detach()截断梯度传递； 一个非常有用的命令
Pytorch安装踩坑之路
cj1561435010的博客
05-20 3189
我擦，Pytorch环境的搭建不得不说，真的有点烦，可能是自己菜吧。 1、选对正确的cuda版本        首先，先确保如果想下载pytorch1.5版本或以上的并且想要使用GPU加速的，确保CUDA版本在10.1及以上，10.0官网已经不支持了。如下图：        现在只需要安装对应版本的cuda就行了，安装pytorch的时候会自动下载适配的cudnn
YOLOv5报错Caught error in DataLoader worker process 1.
最新发布
sinat_56727392的博客
05-13 1143
调小batch-size为2即可。
FaceBoxes.PyTorch代码应用到自己数据遇到的坑
qq_42946328的博客
09-29 1374
1. AssertionError: Invalid device id ngpu的default值改成1 2. FileNotFoundError: [Errno 2] No such file or directory:’./data/WIDER_FACE\annotations\61_Street_Battle_streetfight_61_894.xml’ 将此处的路径改成绝对路径，并且将/改成\\ 3. 查无xml文件 此时需要根据自己的数据，制作img_list.txt文件，文件种每行为图片
IndexError: Caught IndexError in DataLoader worker process 3.
weixin_46141646的博客
07-11 1949
原因分析：报错的原因让人很意想不到，在定义Dataset的时候，是读取.txt文件中的文件名进行数据划分的，在.txt文件中不小心多打了两行空行，这里也会将空行读进去，变成None，导致DataLoader在取数据的时候出问题。解决方法：将空行删去即可。...
centernet-训练和验证自己的数据集(安全帽)
星空点点
07-15 2674
本文以安全帽检测为例，验证下训练自己的数据集 1. 创建json数据 将自己的coco类型的数据放到根目录的dataset目录下 ~/AI/CenterNet-master/data/helmet$ ls annotations images kmeans.py meanvalue.txt /annotations$ ls instances_test2019.json instances_train2019.json instances_val2019.json annotation
解决：TypeError: Caught TypeError in DataLoader worker process 0.
咖乐部
10-07 2万+
部分源码： 报错如下： 网上有帖子说是多进程的问题，要将 num_works 修改为0，其实个人感觉这个error不管多进程的事。 关于多进程报错 可以参见我另一篇文章： 【pytorch】RuntimeError: Integer division of tensors using div or / is no longer supported【解决】。 这里经过检查修改，我是出错在：torchvision.transforms.Totensor() 没有加括号 加上之后，错误解.
h5py读取h5文件报错：Unable to open object (bad heap free list)
chenzhiwen1998的博客
01-30 5181
前言 最近在复现一篇论文的代码时发现了一个致命问题，研究了一天，终于还是破解了。网上找了半天，一个类似的解决方案都没有，记录一下。 错误出现的代码位置： self.feature_app_file = h5py.File(self.opt.input_app_feature, 'r', driver='core') 错误信息 Traceback (most recent call last): File "train.py", line 259, in <module> train
Python迭代DataLoader时出现TypeError: Caught TypeError in DataLoader worker process 0.错误。
热门推荐
weixin_45093926的博客
11-30 9万+
迭代DataLoader时出现TypeError: Caught TypeError in DataLoader worker process 0.错误。遇见一个难以解决的问题 遇见一个难以解决的问题 迭代 DataLoader时出现以下错误，暂时不知道怎么解决，向大家求救，是一个比较稀罕的错误，也分享给大家一个奇葩的问题一起讨论。 Traceback (most recent call last...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

山楂橙子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值