训练时losses损失数组没有数据的情况

[hongtaofly]

在第一个epoch中，已经加载了100组图片并且计算出了损失，但是中间缺 出错：assert losses.numel() != 0。打印了一下发现，当时的损失数组内部没有数据，请问这是什么原因呢？如您回答，不胜感激

[xiaobai111111223]

在第一个epoch中，已经加载了100组图片并且计算出了损失，但是中间缺 出错：assert losses.numel() != 0。打印了一下发现，当时的损失数组内部没有数据，请问这是什么原因呢？如您回答，不胜感激

请问您解决了吗？

[hongtaofly]

youtubb数据集出现了问题，替换了数据集之后错误消失

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2022年04月18日 17:10 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.******@***.***> | | 主题 | Re: [hqucv/siamban] 训练时losses损失数组没有数据的情况 (Issue #64) | 在第一个epoch中，已经加载了100组图片并且计算出了损失，但是中间缺 出错：assert losses.numel() != 0。打印了一下发现，当时的损失数组内部没有数据，请问这是什么原因呢？如您回答，不胜感激 请问您解决了吗？ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[yeluohuazhifou]

计算iou_loss只计算正样本的loss，当前批次没有正样本是就会报错