最新更新:
新版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