shaoheshaohe 发表于 2020-5-22 16:28:22

keras data generator 读数据阻塞

keras封装的太多,

其中,data generator部分,用的语句是
self.queue.get(block=True).get()

这个语句,因为是阻塞模式,很容易堵死数据。

特别是在开多线程时。

所以,如果各位发现在读数据环节,keras死在那里不动了。那么,

将data generator的multiprocessing一定要改为False。

血泪教训。

其实,建议无脑处理,一律将multiprocessing改为False,最好。

最后说,希望大家用pytorch
页: [1]
查看完整版本: keras data generator 读数据阻塞