From 6057496409229437f7532a35971c72476effe696 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 30 Jun 2019 09:57:06 -0700 Subject: [PATCH] Register dpn107 --- timm/models/dpn.py | 1 + 1 file changed, 1 insertion(+) 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(