Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 9, 2022
1 parent b7bc7f5 commit 6575189
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions yolort/models/box_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,7 @@ def __init__(self, in_channels: List[int], num_anchors: int, strides: List[int],
self.num_outputs = num_classes + 5 # number of outputs per anchor
self.strides = strides

head = nn.ModuleList(
nn.Conv2d(ch, self.num_outputs * self.num_anchors, 1) for ch in in_channels
)
head = nn.ModuleList(nn.Conv2d(ch, self.num_outputs * self.num_anchors, 1) for ch in in_channels)

# Initialize biases into head
for mi, s in zip(head, self.strides):
Expand Down

0 comments on commit 6575189

Please sign in to comment.