Register dpn107

5 years ago · 6057496409
parent 3d1a66b6fc
commit 6057496409
1 changed files with 1 additions and 0 deletions
--- 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(