fit() 与 train_on_batch() 是模型训练中最常用的两种方法,自定义训练循环使用相对较少。
fit()
train_on_batch()
fit()的方式是一次把训练数据全部加载到内存中,然后每次批处理batch_size个数据来更新模型参数。这种训练方式适合训练数据量比较小的情况下使用。
train_on_batch()函数接受单批数据,执行反向传播,然后更新模型参数,该批数据的大小可以是任意的,即,它不需要提供明确的批量大小,属于精细化控制训练模型,大部分情
京公网安备 11010502049817号