fix segmentation_head kernel size

brianhou0208 · brianhou0208 · commit e0548a16d0f2 · 2024-10-13T02:39:55.000+08:00
diff --git a/segmentation_models_pytorch/decoders/segformer/model.py b/segmentation_models_pytorch/decoders/segformer/model.py
@@ -74,7 +74,7 @@ def __init__(
             in_channels=decoder_segmentation_channels,
             out_channels=classes,
             activation=activation,
-            kernel_size=3,
+            kernel_size=1,
             upsampling=4,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ def __init__(`
`74`	`74`	`in_channels=decoder_segmentation_channels,`
`75`	`75`	`out_channels=classes,`
`76`	`76`	`activation=activation,`
`77`		`- kernel_size=3,`
	`77`	`+ kernel_size=1,`
`78`	`78`	`upsampling=4,`
`79`	`79`	`)`
`80`	`80`