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 = {