From 63961b36a2e77a8a5a42808f8db85069f6020dfa Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 20 Jun 2019 23:34:20 -0700 Subject: [PATCH] Missed pnasnet entrypoint --- timm/models/pnasnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/pnasnet.py b/timm/models/pnasnet.py index 3f289580..76a63590 100644 --- 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" `_ paper.