Skip to content

Commit 63961b3

Browse files
committed
Missed pnasnet entrypoint
1 parent 6cc214b commit 63961b3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/pnasnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,7 @@ def forward(self, input):
385385
return x
386386

387387

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

0 commit comments

Comments
 (0)