efficient_weight/models/__init__.py

from .resnet import *
from .preact_resnet import *
from .wide_resnet import *
from .small_cnn import *
from .mobilenetv2 import * 
from .wideresnetwithswish import *