Corrected docstring typo of class PatchEmbed
This commit is contained in:
parent
150e5812c7
commit
c1910835a3
@ -380,7 +380,7 @@ class PatchEmbed(nn.Module):
|
|||||||
stride (Tuple): stride of the projection layer.
|
stride (Tuple): stride of the projection layer.
|
||||||
padding (Tuple): padding size of the projection layer.
|
padding (Tuple): padding size of the projection layer.
|
||||||
in_chans (int): Number of input image channels.
|
in_chans (int): Number of input image channels.
|
||||||
embed_dim (int): embed_dim (int): Patch embedding dimension.
|
embed_dim (int): Patch embedding dimension.
|
||||||
"""
|
"""
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user