github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 01:28:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

flake8 style (#352)

Browse Source
This commit is contained in:
Liang Bowen
2022-03-09 17:34:43 +08:00
committed by Frank Lee
parent 54ee8d1254
commit 7eb87f516d
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/nn/layer/vanilla/layers.py
View File

@@ -101,7 +101,7 @@ class WrappedDropPath(nn.Module):
@LAYERS.register_module
class VanillaPatchEmbedding(nn.Module):
"""
"""
2D Image to Patch Embedding
:param img_size: image size