Missed pnasnet entrypoint

6 years ago · 63961b36a2
parent 6cc214bd7a
commit 63961b36a2
1 changed files with 1 additions and 1 deletions
--- a/timm/models/pnasnet.py
+++ b/timm/models/pnasnet.py
@ -385,7 +385,7 @@ class PNASNet5Large(nn.Module):
        return x


-def pnasnet5large(pretrained='imagenet', num_classes=1000, in_chans=3, **kwargs):
+def pnasnet5large(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
    r"""PNASNet-5 model architecture from the
    `"Progressive Neural Architecture Search"
    <https://arxiv.org/abs/1712.00559>`_ paper.