Increase ratio of convnext from 3 to 4.
This commit is contained in:
parent
9a72567b7f
commit
a424a73881
@ -1673,7 +1673,7 @@ class ConvNeXt(nn.Module):
|
|||||||
"""
|
"""
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
channels: int,
|
channels: int,
|
||||||
hidden_ratio: int = 3,
|
hidden_ratio: int = 4,
|
||||||
layerdrop_prob: FloatLike = None):
|
layerdrop_prob: FloatLike = None):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
kernel_size = 7
|
kernel_size = 7
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user