在数据处理和网络定义完成后,跑模型时突然出现了错误:
![7239122-e40ab1387501c3a2.png](https://i-blog.csdnimg.cn/blog_migrate/8231e0cfcf0b283cf8420080ff2b1b94.webp?x-image-process=image/format,png)
OOM
刚开始也不知道哪里的问题,发现有可能是内存耗尽了,然后就放进去500张图片进行fit,然后问题就消失了,猜想应该是数据太大,内存开销不够。
发现官方文档中说可以使用fit_generator()分批训练。
官方文档如下:
fit_generator(self, generator,
steps_per_epoch=None,
epochs=1,
verbose=1,
callbacks=None,
validation_data=None,
validation_steps=None,