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
文件列表
checkmate-master.zip
(预估有个43文件)
checkmate-master
.github
workflows
pythonpackage.yml
2KB
checkmate
poet
power_computation.py
7KB
poet_solver.py
7KB
plot
graph_plotting.py
3KB
暂无评论