分类：fit()

2021
07-04

浅谈Keras中fit()和fit_generator()的区别及其参数的坑

1、fit和fit_generator的区别首先Keras中的fit()函数传入的x_train和y_train是被完整的加载进内存的,当然用起来很方便，但是如果我们数据量很大，那么是不可能将所有数据载入内存的，必将导致内存泄漏，这时候我们可以用fit_generator函数来进行训练。下面是fit传参的例子：history=model.fit(x_train,y_train,epochs=10,batch_size=32,validation_split=0.2)这里需要给出epochs和batch_size，epoch是这... 继续阅读 >