Nanodet训练出现ValueError 解决办法

在配置完Nanodet环境之后,准备使用Nanodet训练自己的数据时候,出现了一个ValueError报错,看起来一头雾水,到Github提交Issue发现这个仓库活跃度不是很高,一直没有得到相关帮助。
 

最后就硬着头皮去想,查找相关文件,发现在yml配置文件中,device设置中batch_size 为160, 我的训练数据集为150,会不会是因为这个原因导致的。

改小了batch_size之后,确保我的训练数据集 / batch_size >=1 ,果然训练成功了。这是一个小的隐形bug, 我们可能开始会用很少一部分数据来测试是否能运行demo, 然后才会正式训练,加上我配置较为简单,一张1050tI 单卡,配置文件里面的device是有多GPU的。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值