From 23703c9dd655cf0ab134e900874cea1ef7056fbb Mon Sep 17 00:00:00 2001 From: Sze-qq <68757353+Sze-qq@users.noreply.github.com> Date: Tue, 18 Oct 2022 20:05:47 +0800 Subject: [PATCH] [NFC] polish colossalai/nn/metric/_utils.py code style (#1727) --- colossalai/nn/metric/_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/colossalai/nn/metric/_utils.py b/colossalai/nn/metric/_utils.py index d4a69f943..eac591b64 100644 --- a/colossalai/nn/metric/_utils.py +++ b/colossalai/nn/metric/_utils.py @@ -1,5 +1,6 @@ import torch + def calc_acc(logits, targets): preds = torch.argmax(logits, dim=-1) correct = torch.sum(targets == preds)