Keras 大坑解決之：fit_generator 停不下來的問題

場景 今天下午用 vgg-16 來訓練 mnist 網絡： 50000 個 sample 1 個 epoch batchsize = 64 使用 2 個 gpu 1080 ti顯卡，跑了一下午就是停不下來了 問題呈現 看這裏該死的 unknown，一眼望不到頭，訓練了 20多萬個 step 就是停不下來 原因分析 因爲使用的 fit_generator 來訓練，這本質是一個生成器，所以如果不指定每

>>阅读原文<<