2021
07-04
07-04
基于keras中训练数据的几种方式对比(fit和fit_generator)
一、train_on_batchmodel.train_on_batch(batchX,batchY)train_on_batch函数接受单批数据,执行反向传播,然后更新模型参数,该批数据的大小可以是任意的,即,它不需要提供明确的批量大小,属于精细化控制训练模型,大部分情况下我们不需要这么精细,99%情况下使用fit_generator训练方式即可,下面会介绍。二、fitmodel.fit(x_train,y_train,batch_size=32,epochs=10)fit的方式是一次把训练数据全部加载到内存中,然后每次批...
继续阅读 >