首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >【YOLO学习笔记】———Batch Size

【YOLO学习笔记】———Batch Size

作者头像
云帆沧海
发布2024-01-17 17:20:30
发布2024-01-17 17:20:30
1.4K0
举报
文章被收录于专栏:编程杂记编程杂记

1.Batch Size 是指在训练神经网络时,一次提供给模型的数据的数量。

在训练神经网络时,模型需要对整个训练数据集进行训练,但是数据集通常很大,如果一次把整个数据集提供给模型训练,可能导致内存不足或运算时间太长。因此,我们通常将数据集分成若干个 Batch,每次提供一个 Batch 给模型训练。Batch Size 就是指一个 Batch 中数据的数量。

Batch Size 过大或过小都不利于模型的训练。 如果 Batch Size 过大,会导致内存不足;如果 Batch Size 过小,模型可能无法很好地捕捉数据中的规律,影响模型的性能。因此,我们需要调整 Batch Size,使其合适。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2024-01-17,如有侵权请联系 cloudcommunity@tencent.com 删除
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档