1 回答

TA貢獻(xiàn)1835條經(jīng)驗 獲得超7個贊
您需要在 model.fit() 函數(shù)中正確設(shè)置steps_per_epochs和validation_steps參數(shù)。
steps_per_epoch訓(xùn)練紀(jì)元被視為完成之前的批次迭代次數(shù)。如果你有一個固定大小的訓(xùn)練集,你可以忽略它,但如果你有一個巨大的數(shù)據(jù)集,或者如果你正在動態(tài)生成隨機(jī)數(shù)據(jù)增強(qiáng),即如果你的訓(xùn)練集有(生成的)無限大小,它可能會很有用。如果您有時間檢查整個訓(xùn)練數(shù)據(jù)集,我建議您跳過此參數(shù)。validation_steps 類似于steps_per_epoch,但在驗證數(shù)據(jù)集而不是訓(xùn)練數(shù)據(jù)上。如果您有時間檢查整個驗證數(shù)據(jù)集,我建議您跳過此參數(shù)。
validation_steps類似于steps_per_epoch,但在驗證數(shù)據(jù)集上而不是在訓(xùn)練數(shù)據(jù)上。如果您有時間檢查整個驗證數(shù)據(jù)集,我建議您跳過此參數(shù)。
在你的情況下設(shè)置:
批量大小 = 200
每紀(jì)元步數(shù) = 300
驗證步驟 = 50
添加回答
舉報