github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish accuracy_2d.py code style (#1719)

pull/1743/head
Ofey Chan 2022-10-18 09:38:16 +08:00 committed by Frank Lee
parent 8860d37846
commit 7e62af28a0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/nn/metric/accuracy_2d.py
View File

@ -8,6 +8,7 @@ from ._utils import calc_acc
class Accuracy2D(nn.Module): class Accuracy2D(nn.Module):
"""Accuracy for 2D parallelism """Accuracy for 2D parallelism
""" """
def __init__(self): def __init__(self):
super().__init__() super().__init__()