python训练模型、如何得到模型训练总时长_模型训练时间的估算

本文介绍了如何通过已知的训练指标(如global_step/sec, examples/sec, Num examples, Batch size, Num steps和Epoch)来估算模型训练的总时长,特别是对于BERT模型在CPU上的训练。通过计算得出,每个迭代大约需要3小时24分钟,而整个训练过程可能需要34小时7分钟。此外,还提供了一个将秒数转换为时间长度的函数getSpanTimes。" 118366234,9823648,Golang计算ASCII字符串中不同字符个数,"['Golang编程', '字符操作', '数据结构与算法']
摘要由CSDN通过智能技术生成

模型训练时间的估算

昨天群里一个朋友训练一个BERT句子对模型,使用的是CPU来进行训练,由于代码是BERT官方代码,并没有显示训练需要的总时间,所以训练的时候只能等待。他截图发了基本的信息,想知道训练完整个模型需要多久。

最开始跑BERT模型的时候,我也碰到这个问题,当时没有深入研究,

不过后来换成keras,bert4keras等框架,可以直接输出时间,比较直观,也就忽略了这个问题。这次回过头来重新理一理,也算温故知新。

话不多说,截图如下:

基础量

先整理一下截图中已知的量:

global_step/sec = 0.204125

表示每秒训练多少个batch批次;

examples/sec = 3.26599

表示每秒可以训练的样本数量 ;

Num examples = 40127

表示样本的总数量

Batch size = 16

表示一批次有多少个样本;

Num steps = 25079

表示总共有多少步(批次)

Epoch = 10

迭代次数,即所有样本训练的次数

迭代次数Epoch没有在截图里,但是可以这样推算:

Epoch = Num steps * Batch size / Num exa

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值