Update cvt.py

fffffgggg54 · fffffgggg54 · commit 183a5dab0f6e · 2024-05-20T13:02:11.000-07:00
diff --git a/timm/models/cvt.py b/timm/models/cvt.py
@@ -344,7 +344,7 @@ def __init__(
             kernel_size = embed_kernel_size,
             stride = embed_stride,
             padding = embed_padding,
-            norm_layer = input_norm_layer,
+            norm_layer = embed_norm_layer,
         )
         self.embed_drop = nn.Dropout(proj_drop)
         

Original file line number	Diff line number	Diff line change
`@@ -344,7 +344,7 @@ def __init__(`
`344`	`344`	`kernel_size = embed_kernel_size,`
`345`	`345`	`stride = embed_stride,`
`346`	`346`	`padding = embed_padding,`
`347`		`- norm_layer = input_norm_layer,`
	`347`	`+ norm_layer = embed_norm_layer,`
`348`	`348`	`)`
`349`	`349`	`self.embed_drop = nn.Dropout(proj_drop)`
`350`	`350`