From b506f7bc367f79c43501a6fca48f9d68621ea420 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Thu, 18 Feb 2021 12:22:12 +0100 Subject: [PATCH] Add a second example model cc @sgugger --- timm/models/dpn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/timm/models/dpn.py b/timm/models/dpn.py index ac9c7755..bd54b7c3 100644 --- a/timm/models/dpn.py +++ b/timm/models/dpn.py @@ -39,7 +39,9 @@ default_cfgs = { url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/dpn68b_ra-a31ca160.pth', mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD), 'dpn92': _cfg( - url='https://github.com/rwightman/pytorch-dpn-pretrained/releases/download/v0.1/dpn92_extra-b040e4a9b.pth'), + url='https://github.com/rwightman/pytorch-dpn-pretrained/releases/download/v0.1/dpn92_extra-b040e4a9b.pth', + hf_revision="main", hf_model_id="julien-c/timm-dpn92", + ), 'dpn98': _cfg( url='https://github.com/rwightman/pytorch-dpn-pretrained/releases/download/v0.1/dpn98-5b90dec4d.pth'), 'dpn131': _cfg(