会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
会飞的两棵树
博客园
首页
新随笔
联系
订阅
管理
CUB reduce errorinvalid configuration argument
解决CUB reduce errorinvalid configuration argument问题
在写TensorFlow代码时遇到报错
CUB reduce errorinvalid configuration argument
在网上查阅说是GPU利用率不高,因为我的代码有一段计算nms的特别耗时,尝试各种方法未能解决,包括将GPU代码移到CPU运算,减少计算量。
就在要放弃的时候头脑发热给损失函数乘了一个因子减小损失。居然成功了,因为我是多任务操作,考虑是损失不平衡,但是我并不确定是不是真的这个原因。
posted @
2019-06-19 09:52
会飞的两棵树
阅读(
1207
) 评论(
0
)
编辑
收藏
举报
会员力量,点亮园子希望
刷新页面
返回顶部
公告