分类：train_on_batch

2020
10-08

Keras之fit_generator与train_on_batch用法

关于Keras中，当数据比较大时，不能全部载入内存，在训练的时候就需要利用train_on_batch或fit_generator进行训练了。两者均是利用生成器，每次载入一个batch-size的数据进行训练。那么fit_generator与train_on_batch该用哪一个呢？train_on_batch(self,x,y,class_weight=None,sample_weight=None)fit_generator(self,generator,samples_per_epoch,nb_epoch,verbose=1,callbacks=[],validation_data=None,nb_val_sample... 继续阅读 >