We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent da8b79b commit 4841933Copy full SHA for 4841933
mmseg/models/decode_heads/stdc_head.py
@@ -37,7 +37,7 @@ def losses(self, seg_logit, seg_label):
37
# parameters. However, it is a constant in original repo and other
38
# codebase because it would not be added into computation graph
39
# after threshold operation.
40
- seg_label = seg_label.float()
+ seg_label = seg_label.to(self.laplacian_kernel)
41
boundary_targets = F.conv2d(
42
seg_label, self.laplacian_kernel, padding=1)
43
boundary_targets = boundary_targets.clamp(min=0)
0 commit comments