[Keras 模型训练] Thread Safe Generator

        最近,在玩语义分割的模型。利用GPU训练的时候,每次跑几个epochs之后,程序崩溃,输出我说我的generator不是线程安全的。查看 trace back发现model.fit_generator在调用自己写的generator出现问题,需要将自己的generator写成线程安全的。          参考keras的#1638 issue找到解决方案。如下图,需要在代码中添加
相关文章
相关标签/搜索