Skip to content

关于评估代码与原始代码有些不同,请大佬帮忙解释一下疑惑? #4

@swjtulinxi

Description

@swjtulinxi

在原始代码的评估.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()
被注释掉了,能不能解释一下啊

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions