fix torchfix errors x2

isaaccorley · isaaccorley · commit c8e500a1bba1 · 2025-12-01T12:18:03.000-06:00
diff --git a/segmentation_models_pytorch/decoders/deeplabv3/decoder.py b/segmentation_models_pytorch/decoders/deeplabv3/decoder.py
@@ -105,7 +105,7 @@ def __init__(
         )
 
         scale_factor = 4 if output_stride == 16 and encoder_depth > 3 else 2
-        self.up = nn.UpsamplingBilinear2d(scale_factor=scale_factor)
+        self.up = nn.Upsample(mode="bilinear", scale_factor=scale_factor)
 
         highres_in_channels = encoder_channels[2]
         highres_out_channels = 48  # proposed by authors of paper
diff --git a/segmentation_models_pytorch/losses/_functional.py b/segmentation_models_pytorch/losses/_functional.py
@@ -226,7 +226,7 @@ def wing_loss(
     idx_smaller = diff_abs < width
     idx_bigger = diff_abs >= width
 
-    loss[idx_smaller] = width * torch.log(1 + diff_abs[idx_smaller] / curvature)
+    loss[idx_smaller] = width * torch.log1p(diff_abs[idx_smaller] / curvature)
 
     C = width - width * math.log(1 + width / curvature)
     loss[idx_bigger] = loss[idx_bigger] - C

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ def __init__(`
`105`	`105`	`)`
`106`	`106`
`107`	`107`	`scale_factor = 4 if output_stride == 16 and encoder_depth > 3 else 2`
`108`		`- self.up = nn.UpsamplingBilinear2d(scale_factor=scale_factor)`
	`108`	`+ self.up = nn.Upsample(mode="bilinear", scale_factor=scale_factor)`
`109`	`109`
`110`	`110`	`highres_in_channels = encoder_channels[2]`
`111`	`111`	`highres_out_channels = 48 # proposed by authors of paper`