model.fit_generator

gpu内存不足使用keras时,使用model.fit_generator.大致分为2种情况

1.已知文件路径需读取

思路: generator 每次生成读取的文件名,然后load数据。

2.已经预读所有数据

思路:生成一个随机数序列,设计生成器每次输出索引序列

你可能感兴趣的:(model.fit_generator)