From d8482a41852562461142c5fef34d022ddbba3c70 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Sat, 12 May 2018 22:49:34 +0100 Subject: [PATCH] add ref --- models/DPN.py | 6 +++++- models/InceptionResNetV2.py | 4 ++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/models/DPN.py b/models/DPN.py index 602328f..3f771f3 100644 --- a/models/DPN.py +++ b/models/DPN.py @@ -16,8 +16,12 @@ many ideas from another PyTorch implementation https://github.com/oyam/pytorch-D This implementation is compatible with the pretrained weights from cypw's MXNet implementation. """ - #__all__ = ['DPN', 'dpn68', 'dpn68b', 'dpn92', 'dpn98', 'dpn131', 'dpn107'] + + +""" +Code imported from https://github.com/Cadene/pretrained-models.pytorch +""" __all__ = ['DPN'] pretrained_settings = { diff --git a/models/InceptionResNetV2.py b/models/InceptionResNetV2.py index 9045227..f84e604 100644 --- a/models/InceptionResNetV2.py +++ b/models/InceptionResNetV2.py @@ -7,6 +7,10 @@ import torch.utils.model_zoo as model_zoo import os import sys +""" +Code imported from https://github.com/Cadene/pretrained-models.pytorch +""" + __all__ = ['InceptionResNetV2'] pretrained_settings = {