MONeT:用于减少DNN训练的内存消耗的MONeT框架

MONeT：深度网络的内存优化 MONeT计划是通过PyTorch实施的，可以在有限的内存预算下以最少的计算开销来训练深度网络。 MONeT共同确定检查点和操作员实施方案，从而将GPU内存减少多达3倍，而计算开销为9-16％。 深度网络的内存优化Aashaka Shah，吴朝元，Jayashree Mohan，Vijay Chidambaram，PhilippKrähenbühl 安装 MONeT已通过PyTorch 1.5.1，torchvision 0.6.1和cudatoolkit 10.1进行了测试。使用python 3.7或更高版本创建一个conda环境。在环境内部，安装以下软件包： cvxpy ， gurobi ， pandas ， ninja-build ， coinor-cbc ， coinor-libcbc-dev和cylp 。 提供安装脚本。 克隆此仓库并安装软件包。