主要介绍了浅谈keras2 predict和fit_generator的坑，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

1、使用predict时，必须设置batch_size,否则效率奇低。 查看keras文档中，predict函数原型： predict(self, x, batch_size=32, verbose=0) 说明： 只使用batch_size=32，也就是说每次将batch_size=32的数据通过PCI总线传到GPU，然后进行预测。在一些问题中，batch_size=32明显是非常小的。而通过PCI传数据是非常耗时的。 所以，使用的时候会发现预测数据时效率奇低，其原因就是batch_size