mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Missed pnasnet entrypoint

Browse Source
This commit is contained in:
Ross Wightman 2019-06-20 23:34:20 -07:00
parent 6cc214bd7a
commit 63961b36a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/pnasnet.py
View File

@ -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.