micro-batch 微批次
显存一次真正容纳的样本组。使用梯度累积时,多个 micro-batch 的梯度合起来才执行一次参数更新。
把一轮数据切成一批批送入模型。改变样本数与批大小,观察更新次数如何改变,也看看最后凑不满的一批究竟会怎样。
micro-batch 微批次显存一次真正容纳的样本组。使用梯度累积时,多个 micro-batch 的梯度合起来才执行一次参数更新。
step 更新步通常指优化器执行一次 optimizer.step()。若累积 4 个微批次,它们合计才算 1 个更新 step。
epoch 训练轮训练流程把计划使用的数据遍历一遍。尾批次是否丢弃,会改变每轮真正被模型看见的样本数。
若 N=1000、B=128 且保留尾批次,一轮有几次更新?尾批次多大?
8 次更新;前 7 批各 128 个样本,尾批次有 104 个样本。
micro-batch=16,累积 4 次后更新参数,有效批大小是多少?
若只考虑单设备且不做其他并行,有效批大小为 16×4=64;4 个微批次对应 1 个优化器 step。