mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-29 05:26:21 +00:00
[NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714)
This commit is contained in:
@@ -8,6 +8,7 @@ from ._utils import calc_acc
|
|||||||
class Accuracy2p5D(nn.Module):
|
class Accuracy2p5D(nn.Module):
|
||||||
"""Accuracy for 2p5D parallelism
|
"""Accuracy for 2p5D parallelism
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user