diff --git a/timm/models/dpn.py b/timm/models/dpn.py index 3a7500cb..1496a067 100644 --- a/timm/models/dpn.py +++ b/timm/models/dpn.py @@ -311,6 +311,7 @@ def dpn131(pretrained=False, num_classes=1000, in_chans=3, **kwargs): return model +@register_model def dpn107(pretrained=False, num_classes=1000, in_chans=3, **kwargs): default_cfg = default_cfgs['dpn107'] model = DPN(