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