CUB reduce errorinvalid configuration argument

解决CUB reduce errorinvalid configuration argument问题

在写TensorFlow代码时遇到报错

CUB reduce errorinvalid configuration argument

在网上查阅说是GPU利用率不高，因为我的代码有一段计算nms的特别耗时，尝试各种方法未能解决，包括将GPU代码移到CPU运算，减少计算量。

就在要放弃的时候头脑发热给损失函数乘了一个因子减小损失。居然成功了，因为我是多任务操作，考虑是损失不平衡，但是我并不确定是不是真的这个原因。