gpu内存不足使用keras时,使用model.fit_generator.大致分为2种情况
1.已知文件路径需读取
思路: generator 每次生成读取的文件名,然后load数据。
2.已经预读所有数据
思路:生成一个随机数序列,设计生成器每次输出索引序列
gpu内存不足使用keras时,使用model.fit_generator.大致分为2种情况
1.已知文件路径需读取
思路: generator 每次生成读取的文件名,然后load数据。
2.已经预读所有数据
思路:生成一个随机数序列,设计生成器每次输出索引序列