gpu內(nèi)存不足使用keras時(shí),使用model.fit_generator.大致分為2種情況
1.已知文件路徑需讀取
思路: generator 每次生成讀取的文件名,然后load數(shù)據(jù)欠气。
2.已經(jīng)預(yù)讀所有數(shù)據(jù)
思路:生成一個隨機(jī)數(shù)序列,設(shè)計(jì)生成器每次輸出索引序列
gpu內(nèi)存不足使用keras時(shí),使用model.fit_generator.大致分為2種情況
1.已知文件路徑需讀取
思路: generator 每次生成讀取的文件名,然后load數(shù)據(jù)欠气。
2.已經(jīng)預(yù)讀所有數(shù)據(jù)
思路:生成一個隨機(jī)數(shù)序列,設(shè)計(jì)生成器每次輸出索引序列