没有合适的资源?快使用搜索试试~ 我知道了~
浅谈keras2 predict和fit_generator的坑
0 下载量 166 浏览量
2020-12-17
07:34:59
上传
评论
收藏 61KB PDF 举报
温馨提示
试读
2页
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总线之间的数据
资源推荐
资源评论
资源评论
weixin_38727087
- 粉丝: 6
- 资源: 966
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功