原始T5论文中的‘step’指什么?是否等同于epochs或iterations?
关于T5论文中“steps”的含义解析
你猜的没错,T5论文里提到的steps就是单个epoch内的迭代次数,严格来说是模型完成一次参数更新的次数,和epochs完全不是一回事:
step(步):指模型用一批训练数据(batch)完成一次前向传播、损失计算、反向传播并更新参数的完整流程,每完成一次就是一个step。epoch(轮次):指模型把整个训练数据集完整遍历并训练一遍的次数。比如C4数据集体量极大,一个epoch包含的steps数量是「C4总样本数 ÷ 训练时设置的batch size」。
论文里说预训练524,288个steps,意思是模型在C4上完成了524,288次参数更新,而不是把数据集过了524k遍——以C4的规模,这个steps数对应的实际epoch数其实非常少。
内容的提问来源于stack exchange,提问作者Jet Lee
相关产品推荐
相关产品推荐

