在原始代码的评估.py代码中此处:
confusion_matrix = torch.from_numpy(confusion_matrix).contiguous().cuda()
confusion_matrix = engine.all_reduce_tensor(confusion_matrix, norm=False).cpu().numpy()
pos = confusion_matrix.sum(1)
res = confusion_matrix.sum(0)
tp = np.diag(confusion_matrix)
而你的:
#confusion_matrix = torch.from_numpy(confusion_matrix).contiguous().cuda()
#confusion_matrix = engine.all_reduce_tensor(confusion_matrix, norm=False).cpu().numpy()
被注释掉了,能不能解释一下啊
在原始代码的评估.py代码中此处:
confusion_matrix = torch.from_numpy(confusion_matrix).contiguous().cuda()
confusion_matrix = engine.all_reduce_tensor(confusion_matrix, norm=False).cpu().numpy()
pos = confusion_matrix.sum(1)
res = confusion_matrix.sum(0)
tp = np.diag(confusion_matrix)
而你的:
#confusion_matrix = torch.from_numpy(confusion_matrix).contiguous().cuda()
#confusion_matrix = engine.all_reduce_tensor(confusion_matrix, norm=False).cpu().numpy()
被注释掉了,能不能解释一下啊