pull/17/head
KaiyangZhou 2018-05-12 22:49:34 +01:00
parent 21e21cdd9d
commit d8482a4185
2 changed files with 9 additions and 1 deletions

View File

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

View File

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