polish dock

2021-02-14 23:04:39 +08:00 · 2021-02-14 23:04:39 +08:00 · 4ab1f506e7
parent 80135d4ef5
commit 4ab1f506e7
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/models/resnet.py
+++ b/python/jittor/models/resnet.py
@ -189,7 +189,7 @@ def Resnet101(pretrained=False, **kwargs):
    Example::

        model = jittor.models.Resnet101()
-        x = jittor.random([10,224,224,3])
+        x = jittor.random([10,3,224,224])
        y = model(x) # [10, 1000]

    """