解决B站视频IndexError: Target 5 is out of bounds.问题

本文介绍如何解决PyTorch模型训练过程中遇到的IndexError问题,并通过调整分类数量及修改全连接层参数来实现模型的正确运行。

再次更新:出现这个问题 大家可以看看自己的数据集在训练集和测试集 我就是的 多了一个空白的文件夹 才导致着也得错误


首先先感谢 这为老哥带给我启发  为照顾和我一样的新手小白 采用喂饭式教程

(5条消息) 解决IndexError: Target 11 is out of bounds.问题_weixin_55191433的博客-优快云博客

在之前的视频中 这位博主的模型一直跑不通 会出现Target 5 is out of bounds 的问题 困扰了我好久在解决完了就有了这个文章 

 在Alex模型中对他进行分类 原模型为

 之后我们将他只改一个值

 就可以正常跑通了 我们之前写的num_classes=1000是不行的 应为我们只分为5类 至于为啥是6而不是5 我确实不知道 后面会进行相关了解 也希望了解的人可以在评论区告诉大家和我

在举一个例子在 Google模型中 我们不仅要把所有的全连接层改了如图

我直接写了一个变量 省的麻烦  但是这里6不行 要一个个试

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值