2020年2月10日
torch 数组按顺序进行编号，用在hard negative mining
摘要： 在hard negative mining中，要负样本的得分进行排序，选取得分最高的参与训练。这就涉及到选取top k个负样本或掩膜，因此要对数组按顺序进行编号。 在pytorch中, 函数可以返回排序好的数组和元素的索引，用两次 函数即可以得到数组元素的编号。 附上一段 "hard example 阅读全文
posted @ 2020-02-10 20:31 骏腾 阅读(866) 评论(0) 推荐(0) 编辑