/Memonger_Densenet_Resnet_MobileNet_LSTM

This project contains a series of python script to give sublinear memory plans of deep neural networks including Densenet, Resnet, Mobilenet-v2 and LSTM. This allows you to trade computation for memory and get sublinear memory cost, so you can train bigger/deeper nets with limited resources.

Primary LanguagePythonMIT LicenseMIT

Watchers