Skip to content

Commit f4ea996

Browse files
authored
Fix ufmt formatting
1 parent ee93c07 commit f4ea996

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

torchvision/models/detection/retinanet.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,13 @@ class RetinaNetHead(nn.Module):
6565
norm_layer (callable, optional): Module specifying the normalization layer to use. Default: None
6666
"""
6767

68-
def __init__(self, in_channels, num_anchors, num_classes, norm_layer: Optional[Callable[..., nn.Module]] = None) -> None:
68+
def __init__(
69+
self,
70+
in_channels,
71+
num_anchors,
72+
num_classes,
73+
norm_layer: Optional[Callable[..., nn.Module]] = None,
74+
) -> None:
6975
super().__init__()
7076
self.classification_head = RetinaNetClassificationHead(
7177
in_channels, num_anchors, num_classes, norm_layer=norm_layer

0 commit comments

Comments
 (0)