colossalai.nn.metric.accuracy_2p5d

class colossalai.nn.metric.accuracy_2p5d.Accuracy2p5D

Accuracy for 2p5D parallelism

forward(logits, targets)

Calculate the accuracy of predicted labels.

Parameters

• logits – Predicted labels

• targets – True labels from data