Skip to content

Commit a1c4c1e

Browse files
committed
Update cvt.py
1 parent c7120f6 commit a1c4c1e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/cvt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -367,8 +367,8 @@ def __init__(
367367
qk_norm: bool = False,
368368
attn_drop: float = 0.,
369369
proj_drop: float = 0.,
370-
input_norm_layer = LayerNorm2d,
371-
norm_layer: nn.Module = LayerNorm,
370+
input_norm_layer = partial(LayerNorm2d, eps=1e-5),
371+
norm_layer: nn.Module = partial(LayerNorm, eps=1e-5),
372372
init_values: Optional[float] = None,
373373
drop_path_rate: float = 0.,
374374
mlp_layer: nn.Module = Mlp,

0 commit comments

Comments
 (0)