From 9e5e95e3b1ccff395854d88fa91d4cb330810b12 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 16 Dec 2020 16:19:12 +0000 Subject: [PATCH 1/2] Fixing incorrect doc example in MNASNet. --- torchvision/models/mnasnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/mnasnet.py b/torchvision/models/mnasnet.py index 3ed496c88d2..69b0696a3e5 100644 --- a/torchvision/models/mnasnet.py +++ b/torchvision/models/mnasnet.py @@ -93,7 +93,7 @@ def _get_depths(alpha: float) -> List[int]: class MNASNet(torch.nn.Module): """ MNASNet, as described in https://arxiv.org/pdf/1807.11626.pdf. This implements the B1 variant of the model. - >>> model = MNASNet(1000, 1.0) + >>> model = MNASNet(1.0, num_classes=1000) >>> x = torch.rand(1, 3, 224, 224) >>> y = model(x) >>> y.dim() From ed1b0c035bd4cdca358c45b18a257d861b95a234 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 16 Dec 2020 16:22:01 +0000 Subject: [PATCH 2/2] Fixing incorrect output. --- torchvision/models/mnasnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/mnasnet.py b/torchvision/models/mnasnet.py index 69b0696a3e5..3c703774dda 100644 --- a/torchvision/models/mnasnet.py +++ b/torchvision/models/mnasnet.py @@ -97,7 +97,7 @@ class MNASNet(torch.nn.Module): >>> x = torch.rand(1, 3, 224, 224) >>> y = model(x) >>> y.dim() - 1 + 2 >>> y.nelement() 1000 """