Skip to content

Commit b5db5e8

Browse files
committed
Fix example in RetinaNet documentation (#2808)
1 parent 643fdc3 commit b5db5e8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/models/detection/retinanet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,8 +312,8 @@ class RetinaNet(nn.Module):
312312
>>> # map could potentially have different sizes and
313313
>>> # aspect ratios
314314
>>> anchor_generator = AnchorGenerator(
315-
>>> sizes=tuple((x, int(x * 2 ** (1.0 / 3)), int(x * 2 ** (2.0 / 3))) for x in [32, 64, 128, 256, 512]),
316-
>>> aspect_ratios=((0.5, 1.0, 2.0),) * 5
315+
>>> sizes=((32, 64, 128, 256, 512),),
316+
>>> aspect_ratios=((0.5, 1.0, 2.0),)
317317
>>> )
318318
>>>
319319
>>> # put the pieces together inside a RetinaNet model

0 commit comments

Comments
 (0)