摘要: _, predited = torch.max(outputs,1) # 此处表示返回一个元组中有两个值,但是对第一个不感兴趣 返回的元组的第一个元素是image data,即是最大的值;第二个元素是label,即是最大的值对应的索引。由于我们只需要label(最大值的索引),所以有 _ , pre 阅读全文
posted @ 2020-10-22 18:49 阿洛萌萌哒 阅读(2307) 评论(0) 推荐(0) 编辑