Hetu/examples/cnn/pytorch_models/__init__.py

from .mlp import mlp
from .resnet import resnet18, resnet34, resnet50
from .vgg import vgg16, vgg19
from .rnn import rnn