[Keras 模型訓練] Thread Safe Generator

        最近,在玩語義分割的模型。利用GPU訓練的時候,每次跑幾個epochs之後,程序崩潰,輸出我說我的generator不是線程安全的。查看 trace back發現model.fit_generator在調用自己寫的generator出現問題,需要將自己的generator寫成線程安全的。          參考keras的#1638 issue找到解決方案。如下圖,需要在代碼中添加
相關文章
相關標籤/搜索