fix torchfix errors

isaaccorley · isaaccorley · commit 31249514d6b9 · 2025-12-01T11:35:29.000-06:00
diff --git a/segmentation_models_pytorch/base/heads.py b/segmentation_models_pytorch/base/heads.py
@@ -10,7 +10,7 @@ def __init__(
             in_channels, out_channels, kernel_size=kernel_size, padding=kernel_size // 2
         )
         upsampling = (
-            nn.UpsamplingBilinear2d(scale_factor=upsampling)
+            nn.Upsample(mode="bilinear", scale_factor=upsampling)
             if upsampling > 1
             else nn.Identity()
         )

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ def __init__(`
`10`	`10`	`in_channels, out_channels, kernel_size=kernel_size, padding=kernel_size // 2`
`11`	`11`	`)`
`12`	`12`	`upsampling = (`
`13`		`- nn.UpsamplingBilinear2d(scale_factor=upsampling)`
	`13`	`+ nn.Upsample(mode="bilinear", scale_factor=upsampling)`
`14`	`14`	`if upsampling > 1`
`15`	`15`	`else nn.Identity()`
`16`	`16`	`)`