diff --git a/timm/models/pnasnet.py b/timm/models/pnasnet.py index dc9b3e20..64d83e3c 100644 --- a/timm/models/pnasnet.py +++ b/timm/models/pnasnet.py @@ -390,7 +390,7 @@ def pnasnet5large(pretrained=False, num_classes=1000, in_chans=3, **kwargs): `_ paper. """ default_cfg = default_cfgs['pnasnet5large'] - model = PNASNet5Large(num_classes=1000, in_chans=in_chans, **kwargs) + model = PNASNet5Large(num_classes=num_classes, in_chans=in_chans, **kwargs) model.default_cfg = default_cfg if pretrained: load_pretrained(model, default_cfg, num_classes, in_chans) diff --git a/timm/models/res2net.py b/timm/models/res2net.py index 134cf00d..8655776c 100644 --- a/timm/models/res2net.py +++ b/timm/models/res2net.py @@ -229,7 +229,7 @@ def res2next50(pretrained=False, num_classes=1000, in_chans=3, **kwargs): default_cfg = default_cfgs['res2next50'] res2net_block_args = dict(scale=4) model = ResNet(Bottle2neck, [3, 4, 6, 3], base_width=4, cardinality=8, - num_classes=1000, in_chans=in_chans, block_args=res2net_block_args, **kwargs) + num_classes=num_classes, in_chans=in_chans, block_args=res2net_block_args, **kwargs) model.default_cfg = default_cfg if pretrained: load_pretrained(model, default_cfg, num_classes, in_chans)