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太小了。
2022-05-06 02:25:00
说明:只使用batch_size=32,也就是说每次将batch_size=32的数据通过PCI总线传到GPU,然后进行预测。在一些问题中,batch_size=32明显是非常小的。而通过PCI传数据是非常耗时的。所以,使用的时候会发现预测数据时效率奇低,其原因就是batch_size太小了。