ModelFeast

Please star ModelFeast if it helps you. This is very important to me! Thanks very much !

ModelFeast is more than model-zoo! It is:

Avalible models

Pretrained models ( trained on ImageNet ) for 2D CNN is now avalible on Baiduyun(fst6) and Google Drive

This part is still on progress. Not avalible to train now, but model architecture can been seen here.

Determine what you need and read corresponding tutorials

Or you can use modelfeast simply via pip !

pip3 install modelfeast

The features are more than you could think of:

Train and save model within 3 lines !
Multi GPU support !
Include the most popular 2D CNN, 3D CNN, and CRNN models !
Allow any input image size (pytorch official model zoo limit your input size harshly) !
Help you sweep all kinds of classification competitions.