操蛋的程序员日常

model.fit_generator

2019-11-01  本文已影响0人  bot张翔宇

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

1.已知文件路径需读取

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

2.已经预读所有数据

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

上一篇 下一篇

猜你喜欢

热点阅读