diff --git a/colossalai/nn/metric/accuracy_2p5d.py b/colossalai/nn/metric/accuracy_2p5d.py index 337c6af4a..3044da065 100644 --- a/colossalai/nn/metric/accuracy_2p5d.py +++ b/colossalai/nn/metric/accuracy_2p5d.py @@ -8,6 +8,7 @@ from ._utils import calc_acc class Accuracy2p5D(nn.Module): """Accuracy for 2p5D parallelism """ + def __init__(self): super().__init__()