keras在compile完模型后需要训练,除了常用的model.fit()与model.fit_generator外还有model.train_on_bantch
作用:对一批样品进行单梯度更新。即对一个epoch中的一个样本进行一次训练
使用train_on_batch优点:
1、更精细自定义训练过程,更精准的收集 loss 和 metrics
2、分步训练模型-GAN的实现
3、多GPU训练保存模型更加方便
深度学习学习笔记——keras中的train_on_batch
最新推荐文章于 2022-10-24 21:37:51 发布