用pytorch训练EfficientNet网络。出现错误IndexError: dimension specified as 0 but tensor has no dimensions

批量大小与样本数量不匹配问题及解决方案
当训练或验证数据集的样本数量与批处理大小不成整数倍时,会导致尺寸错误。该错误表现为'IndexError: dimensions specified as 0 but tensor has no dimensions'。解决方法是调整样本数量和批处理大小,确保它们之间可以整除,以消除报错。可以通过逐步减少批处理大小来找到合适的匹配值。

if input.size(0) != target.size(0):
IndexError: dimension specified as 0 but tensor has no dimensions

原因:train或者validation dataset 样本数量与batchsize不是整除的关系
解决:调整为整除关系,不再报错(从最大batchsize递减慢慢测试出来的)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值