Commit f586887a authored by DouJS's avatar DouJS Committed by Frank Lee
Browse files

[NFC] polish colossalai/nn/layer/colossalai_layer/dropout.py code style (#1568)

parent c7d49329
......@@ -13,6 +13,7 @@ class Dropout(ColossalaiModule):
p (float, optional): probability of an element to be zeroed, defaults 0.5.
inplace (bool, optional): whether to do dropout in-place, default to be False.
"""
def __init__(self, p: float = 0.5, inplace: bool = False) -> None:
tensor_parallel = get_tensor_parallel_mode()
if tensor_parallel == "1d":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment