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

23703c9d · Sze-qq · Frank Lee · 754aa7c8 · 23703c9d
Commit 23703c9d authored Oct 18, 2022 by Sze-qq Committed by Frank Lee Oct 19, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

colossalai/nn/metric/_utils.py colossalai/nn/metric/_utils.py +1 -0

No files found.
--- a/colossalai/nn/metric/_utils.py
+++ b/colossalai/nn/metric/_utils.py
 import torch

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