We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b0c6c01 commit 26c6855Copy full SHA for 26c6855
timm/models/cvt.py
@@ -367,8 +367,8 @@ def __init__(
367
qk_norm: bool = False,
368
attn_drop: float = 0.,
369
proj_drop: float = 0.,
370
- input_norm_layer = LayerNorm2d,
371
- norm_layer: nn.Module = LayerNorm,
+ input_norm_layer = partial(LayerNorm2d, eps=1e-5),
+ norm_layer: nn.Module = partial(LayerNorm, eps=1e-5),
372
init_values: Optional[float] = None,
373
drop_path_rate: float = 0.,
374
mlp_layer: nn.Module = Mlp,
0 commit comments