Skip to content

Commit 8aa3f58

Browse files
committed
Use _reduced_tail=True.
1 parent f8cbe46 commit 8aa3f58

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed
Binary file not shown.

torchvision/models/detection/ssdlite.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,13 +170,14 @@ def ssdlite320_mobilenet_v3_large(pretrained: bool = False, progress: bool = Tru
170170
if pretrained:
171171
pretrained_backbone = False
172172

173-
rescaling = not pretrained_backbone
173+
# Enable [-1, 1] rescaling and reduced tail if no pretrained backbone is selected
174+
rescaling = reduce_tail = not pretrained_backbone
174175

175176
if norm_layer is None:
176177
norm_layer = partial(nn.BatchNorm2d, eps=0.001, momentum=0.03)
177178

178179
backbone = _mobilenet_extractor("mobilenet_v3_large", progress, pretrained_backbone, trainable_backbone_layers,
179-
norm_layer, rescaling, _width_mult=1.0)
180+
norm_layer, rescaling, _reduced_tail=reduce_tail, _width_mult=1.0)
180181

181182
size = (320, 320)
182183
anchor_generator = DefaultBoxGenerator([[2, 3] for _ in range(6)], min_ratio=0.2, max_ratio=0.95)

0 commit comments

Comments
 (0)