mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-23 14:10:29 +00:00
[NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714)
This commit is contained in:
parent
9d0560af9c
commit
e1d780030d
@ -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__()
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user