keras2 predict和fit_generator的坑
来源:互联网 发布:淘宝上井江茶油造假 编辑:程序博客网 时间:2024/06/08 14:26
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时,必须人为设置好batch_size,否则PCI总线之间的数据传输次数过多,性能会非常低下。
2、fit_generator
说明:keras 中 fit_generator参数steps_per_epoch已经改变含义了,目前的含义是一个epoch分成多少个batch_size。旧版的含义是一个epoch的样本数目。
如果说训练样本树N=1000,steps_per_epoch = 10,那么相当于一个batch_size=100,如果还是按照旧版来设置,那么相当于
batch_size = 1,会性能非常低。
经验:必须明确fit_generator参数steps_per_epoch
阅读全文
0 0
- keras2 predict和fit_generator的坑
- 【Keras】训练时显存out of memory的解决办法——fit_generator函数
- libsvm的使用svm-scale和svm-train与svm-predict
- libsvm2.88之“svm-predict”的使用
- 使用python fit,predict出现的错误
- Felomeng翻译:libsvm2.88之“svm-predict”的使用
- Predict the Winner一个动态规划的问题解法详解
- [LeetCode] Predict the Winner 预测得分最高的赢家
- keras的基本用法(五)——图像predict
- keras 迁移学习, 微调, model的predict函数定义
- Future Predict
- Theano predict
- keras2.0 Merge层改为函数式
- 基于opencv中HOG+SVM的车轮识别及predict()函数的改进
- mxnet的Predict with pre-trained models教程,以及一些常见的问题
- 用scikit-learn 训练SVM的例子, clf.fit() clf.predict() 的作用
- caffe finetune predict and classify the lung nodule( 肺结节的分类)
- sklearn中,RandomForest、DecisionTree、cython(spliter、builder)三个层面的fit、predict
- PYTHON smtp 发送qq邮件
- vue中如何使用less
- Sqrt(x)
- 实现文本域输入字数的限制
- pcl::ExtractPolygonalPrismData类
- keras2 predict和fit_generator的坑
- 应用expect的自动应答
- Add to List 313. Super Ugly Number
- Java8的四大核心函数式接口
- 第一课、java开发环境的安装
- 随机一致性检测RANSAC
- 位操作
- Eclipse用tomcat有事出现的问题
- 开课第一天