摘要: 解决CUB reduce errorinvalid configuration argument问题 在写TensorFlow代码时遇到报错 在网上查阅说是GPU利用率不高,因为我的代码有一段计算nms的特别耗时,尝试各种方法未能解决,包括将GPU代码移到CPU运算,减少计算量。 就在要放弃的时候头 阅读全文
posted @ 2019-06-19 09:52 会飞的两棵树 阅读(1207) 评论(0) 推荐(0) 编辑