github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 14:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/nn/metric/_utils.py code style (#1727)

Browse Source
This commit is contained in:
Sze-qq 2022-10-18 20:05:47 +08:00 committed by Frank Lee
parent 754aa7c81f
commit 23703c9dd6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/nn/metric/_utils.py
View File

@ -1,5 +1,6 @@
import torch import torch
def calc_acc(logits, targets): def calc_acc(logits, targets):
preds = torch.argmax(logits, dim=-1) preds = torch.argmax(logits, dim=-1)
correct = torch.sum(targets == preds) correct = torch.sum(targets == preds)