Skip to content

Commit bbcdbc0

Browse files
committed
make style
1 parent 849cc03 commit bbcdbc0

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

examples/community/hd_painter.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -900,13 +900,13 @@ class GaussianSmoothing(nn.Module):
900900
in the input using a depthwise convolution.
901901
902902
Args:
903-
channels (`int` or `sequence`):
903+
channels (`int` or `sequence`):
904904
Number of channels of the input tensors. The output will have this number of channels as well.
905-
kernel_size (`int` or `sequence`):
905+
kernel_size (`int` or `sequence`):
906906
Size of the Gaussian kernel.
907-
sigma (`float` or `sequence`):
907+
sigma (`float` or `sequence`):
908908
Standard deviation of the Gaussian kernel.
909-
dim (`int`, *optional*, defaults to `2`):
909+
dim (`int`, *optional*, defaults to `2`):
910910
The number of dimensions of the data. Default is 2 (spatial dimensions).
911911
"""
912912

@@ -953,7 +953,7 @@ def forward(self, input):
953953
Input to apply Gaussian filter on.
954954
955955
Returns:
956-
`torch.Tensor`:
956+
`torch.Tensor`:
957957
The filtered output tensor with the same shape as the input.
958958
"""
959959
return self.conv(input, weight=self.weight.to(input.dtype), groups=self.groups, padding="same")

0 commit comments

Comments
 (0)