赞
踩
最近开发深度学习算法的可交互的演示程序,遇到这样一个需求:实时显示每个批次每个epoch的loss值,那么keras模型的fit函数不太能用了,fit函数是把所有的数据喂到model里面,自然就会想到keras会有分批次训练的函数,不需要自己造轮子了
keras里边有两个和分批次相关的函数:train_on_batch和fit_generator,这里分享一些介绍到关键点的笔记

这里边有一个实际的例子非常好,在链接的第6部分,记得查看啊~

那么,我对fit_generator函数的理解是,它可以分批次的训练,怎么划分是用户自己定义好的,fit_generator自动去调用那个生成器,一次一次的循环。


https://blog.csdn.net/wuxulong123/article/details/105024298






https://blog.csdn.net/weixin_42744102/article/details/90083903

它的工程代码太棒了!

最后忘了了解为什么会有train_on_batch和fit_generator函数,可不是因为我们最初那个可视化的需求,而是,数据规模太大的时候,一次读入内存,内存会泄漏,才有了分批次训练
https://cloud.tencent.com/developer/article/1724587

好啦~对于train_on_batch和fit_generator的学习就到这里啦,欢迎点赞、收藏和转发嗷~
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。