2020
10-08
10-08
浅谈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太小了。经验：使用predict时，必须人为设置好ba...
继续阅读 >