#深入探究# Tensorflow 中 batch_size 的设置问题

在学习tensorflow时,有几个地方都能设置batch_size 参数,这里对batch_size的设置做一个总结
首先强调一下,这个batch_size的作用就是规定每次梯度更新的样本数量(规定小批量梯度下降每次梯度更新的样本数量)

目前已知设置 batch_size 的地方有:

  • tf.data.batch() 函数、生成器或keras.utils.Sequence实例 都能通过内部方法生成batch
  • fit() 函数中的 batch_size 参数

根据输入数据类型的不同,可分为如下两种情况:

  1. 如果fit() 函数的数据采用数据集(包括 tf.data, tf.data默认batch_size=32),生成器或keras.utils.Sequence实例的形式(因为它们会内部生成batch),那么这些数据内部的会生成batch,此时通过 fit() 函数中的 batch_size 参数设置batch_size 无效(fit方法会内部屏蔽 batch_size 参数)

  2. 当如果fit() 函数的数据不是数据集(包括 tf.data, tf.data默认batch_size=32),生成器或keras.utils.Sequence实例的形式时,则可以通过fit() 函数中的 batch_size 参数设置 batch_size

batch()方法不能重复使用,使用batch()方法会增加一个维度!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

energy_百分百

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值