github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-19 18:00:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/nn/layer/vanilla/__init__.py code style (#1293)

Browse Source
This commit is contained in:
Liping233 2022-07-13 11:15:21 +08:00 committed by Frank Lee
parent 87f679aeae
commit 1000a41fd5
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
colossalai/nn/layer/vanilla/__init__.py
View File

@ -1,7 +1,6 @@
from .layers import (DropPath, VanillaClassifier, VanillaLayerNorm, from .layers import (DropPath, VanillaClassifier, VanillaLayerNorm, VanillaPatchEmbedding, WrappedDropout,
VanillaPatchEmbedding, WrappedDropout, WrappedDropPath) WrappedDropPath)
__all__ = [ __all__ = [
"VanillaLayerNorm", "VanillaPatchEmbedding", "VanillaClassifier", "VanillaLayerNorm", "VanillaPatchEmbedding", "VanillaClassifier", "DropPath", "WrappedDropout", "WrappedDropPath"
"DropPath", "WrappedDropout", "WrappedDropPath"
] ]