keras模型中的model.fit()和model.fit_generator()的區別


最新更新:

  新版tf中,model.fit()已經支持加載生成器對象了,故統一使用model.fit()即可

===================================================================

1.兩者的語法如下:

 

 

 

 2. 從上可以看出,fit()是將訓練數據 x 和 y 完整的加載到內存中,如果數據量很大,不可能把所有數據一並加載到內存,必將導致內存泄漏;

而fit_generator() 加載的是一個 生成器,訓練數據是通過該生成器產生的

3. batch_size、epoch、steps_per_epoch的區別

batch_size代表批次,即每次送多少組數據一起訓練,例如batch_size = 32,即每次訓練的數據都是一個 (32, height, width, channels) 4維向量組

epoch代表完整訓練的次數,所有批次全部訓練完一遍,叫做 1 epoch

steps_per_epoch 代表 每次 訓練多少步,其值 = 樣本數 / batch_size


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM