From d3f3be359fefb6c7584fedef3e9b298e56474d7a Mon Sep 17 00:00:00 2001
From: kevinlu1211 <kevinyihchyunlu@gmail.com>
Date: Mon, 5 Aug 2019 09:47:31 +1000
Subject: [PATCH] Update __init__.py

Best practices to import with "*", also fixes my use case of automatically generate lookup dictionary by having the names of the models being exported:

```
from pretrainedmodels.models import __all__ as pretrained_models_names

pretrained_models_lookup = {
    name: getattr(importlib.import_module("pretrainedmodels.models"), name)
    for name in pretrained_models_names
}

lookup = {**pretrained_models_lookup}
```
---
 pretrainedmodels/models/__init__.py | 49 +++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/pretrainedmodels/models/__init__.py b/pretrainedmodels/models/__init__.py
index 83c2392..9dd4690 100644
--- a/pretrainedmodels/models/__init__.py
+++ b/pretrainedmodels/models/__init__.py
@@ -1,4 +1,52 @@
 from __future__ import print_function, division, absolute_import
+
+__all__ = ['fbresnet152',
+ 'cafferesnet101',
+ 'bninception',
+ 'resnext101_32x4d',
+ 'resnext101_64x4d',
+ 'inceptionv4',
+ 'inceptionresnetv2',
+ 'nasnetalarge',
+ 'nasnetamobile',
+ 'alexnet',
+ 'densenet121',
+ 'densenet169',
+ 'densenet201',
+ 'densenet161',
+ 'resnet18',
+ 'resnet34',
+ 'resnet50',
+ 'resnet101',
+ 'resnet152',
+ 'inceptionv3',
+ 'squeezenet1_0',
+ 'squeezenet1_1',
+ 'vgg11',
+ 'vgg11_bn',
+ 'vgg13',
+ 'vgg13_bn',
+ 'vgg16',
+ 'vgg16_bn',
+ 'vgg19_bn',
+ 'vgg19',
+ 'dpn68',
+ 'dpn68b',
+ 'dpn92',
+ 'dpn98',
+ 'dpn131',
+ 'dpn107',
+ 'xception',
+ 'senet154',
+ 'se_resnet50',
+ 'se_resnet101',
+ 'se_resnet152',
+ 'se_resnext50_32x4d',
+ 'se_resnext101_32x4d',
+ 'pnasnet5large',
+ 'polynet'
+]
+
 from .fbresnet import fbresnet152
 
 from .cafferesnet import cafferesnet101
@@ -56,3 +104,4 @@
 
 from .pnasnet import pnasnet5large
 from .polynet import polynet
+