Fix example in RetinaNet documentation (#2808)

fef2c8a5 · Francisco Massa · GitHub · 4db3dc6d · fef2c8a5
Unverified Commit fef2c8a5 authored Oct 14, 2020 by Francisco Massa Committed by GitHub Oct 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/models/detection/retinanet.py torchvision/models/detection/retinanet.py +2 -2

No files found.
--- a/torchvision/models/detection/retinanet.py
+++ b/torchvision/models/detection/retinanet.py
@@ -312,8 +312,8 @@ class RetinaNet(nn.Module):
        >>> # map could potentially have different sizes and
        >>> # aspect ratios
        >>> anchor_generator = AnchorGenerator(
-        >>>     sizes=tuple((x, int(x * 2 ** (1.0 / 3)), int(x * 2 ** (2.0 / 3))) for x in [32, 64, 128, 256, 512]),
-        >>>     aspect_ratios=((0.5, 1.0, 2.0),) * 5
+        >>>     sizes=((32, 64, 128, 256, 512),),
+        >>>     aspect_ratios=((0.5, 1.0, 2.0),)
        >>> )
        >>>
        >>> # put the pieces together inside a RetinaNet model