Skip to content

Commit 6488c41

Browse files
committed
Fix linter
1 parent 24b8643 commit 6488c41

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

torchvision/models/detection/faster_rcnn.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -570,7 +570,9 @@ def fasterrcnn_resnet50_fpn_v2(
570570
backbone = _resnet_fpn_extractor(backbone, trainable_backbone_layers, norm_layer=nn.BatchNorm2d)
571571
rpn_anchor_generator = _default_anchorgen()
572572
rpn_head = RPNHead(backbone.out_channels, rpn_anchor_generator.num_anchors_per_location()[0], conv_depth=2)
573-
box_head = FastRCNNConvFCHead((backbone.out_channels, 7, 7), [256, 256, 256, 256], [1024], norm_layer=nn.BatchNorm2d)
573+
box_head = FastRCNNConvFCHead(
574+
(backbone.out_channels, 7, 7), [256, 256, 256, 256], [1024], norm_layer=nn.BatchNorm2d
575+
)
574576
model = FasterRCNN(
575577
backbone,
576578
num_classes=num_classes,

torchvision/models/detection/mask_rcnn.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -508,7 +508,9 @@ def maskrcnn_resnet50_fpn_v2(
508508
backbone = _resnet_fpn_extractor(backbone, trainable_backbone_layers, norm_layer=nn.BatchNorm2d)
509509
rpn_anchor_generator = _default_anchorgen()
510510
rpn_head = RPNHead(backbone.out_channels, rpn_anchor_generator.num_anchors_per_location()[0], conv_depth=2)
511-
box_head = FastRCNNConvFCHead((backbone.out_channels, 7, 7), [256, 256, 256, 256], [1024], norm_layer=nn.BatchNorm2d)
511+
box_head = FastRCNNConvFCHead(
512+
(backbone.out_channels, 7, 7), [256, 256, 256, 256], [1024], norm_layer=nn.BatchNorm2d
513+
)
512514
mask_head = MaskRCNNHeads(backbone.out_channels, [256, 256, 256, 256], 1, norm_layer=nn.BatchNorm2d)
513515
model = MaskRCNN(
514516
backbone,

0 commit comments

Comments
 (0)