首先,我们知道要划分数据集,通常分为训练集,验证集,测试集。

在训练模型的时候,从训练集中一个一个取出来训练效率太慢,可以一次取一批来进行训练,batch_size就是这一批数据的数量设置。

1个epoch就是把训练集里面的所有数据都训练一遍。

模型的迭代次数就是训练集所有数据量除以batch_size,比如有100个数据,batch_size=10,那么迭代次数就是100÷10=10(次)。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐