checkmate:在TensorFlow 2.0中训练神经网络内存减少5倍 源码

看论文! checkmate使研究人员能够训练不适合GPU内存的大型最新模型,从而打破了GPU内存的checkmate 。 Checkmate应用最佳张量重新实现(如我们在MLSys 2020中的文章中详细介绍)以权衡空间和时间。 目前,Checkmate仅支持TensorFlow 2.0。 PyTorch支持即将推出! 安装 将死取决于: ,即pip install tensorflow或pip install tensorflow-gpu 。 在Debian Linux / Ubuntu上安装CyLP $ sudo apt install coinor-cbc coinor-l