colossalai.nn.metric.accuracy_2p5d

class colossalai.nn.metric.accuracy_2p5d.Accuracy2p5D[source]

Accuracy for 2p5D parallelism

forward(logits, targets)[source]

Calculate the accuracy of predicted labels.

Parameters

logits (torch.tensor) – Predicted labels.
targets (torch.tensor) – True labels from data.

Returns

the accuracy of prediction.

Return type

float