开发者社区> 问答> 正文

为什么移动框架中loss还需要乘batch_size呢?

已解决

为什么移动框架中loss还需要乘batch_size呢?

展开
收起
游客dndzx3dbvf2jq 2022-09-16 22:12:28 1736 0
1 条回答
写回答
取消 提交回答
  • 推荐回答

    比如这个语句: train_loss += loss.item() * images.size(0) ,为什么总的损失train_loss不直接累加loss.item(),还在后边乘上images.size(0)呢?

    loss.item()应该是一个batch size的平均损失,×images.size(0)那就是一个batch size的总损失,所以train_loss很可能是求一个epoch的loss之和。

    2022-09-16 22:33:28
    赞同 展开评论 打赏
来源圈子
更多
收录在圈子:
+ 订阅
技术图谱:由专家组参与技术图谱的绘制与编写,知识与实践的结合让开发者们掌握学习路线与逻辑,快速提升技能 电子书:电子书由阿里内外专家打造,供开发者们下载学习,更与课程相结合,使用户更易理解掌握课程内容 训练营:学习训练营 深入浅出,专家授课,带领开发者们快速上云 精品课程:汇集知识碎片,解决技术难题,体系化学习场景,深入浅出,易于理解 技能自测:提供免费测试,摸底自查 体验实验室:学完即练,云资源免费使用
问答排行榜
最热
最新

相关电子书

更多
Processing Terabyte Scale Genomics Datasets with ADAM 立即下载
Spark ML with High Dimensional Labels 立即下载
Random Walks on Large Scale Graphs with Spark 立即下载