文件名称：checkmate：在TensorFlow 2.0中训练神经网络，内存减少5倍
文件大小：61KB
文件格式：ZIP
更新时间：2024-02-21 07:42:50
python distributed-systems deep-learning compiler memory
看论文！ 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-libcbc-dev $ pip install cylp 在MacOS上安装CyLP 设置CyLP的最简单方法是使用。 $ brew tap coin-or-tools/coinor $ brew install coin-or-tools/coinor/cbc pkg-config $ pip install cylp 一旦安装了TensorFlow 2.0和CyLP，就可以通过pip
【文件预览】：
checkmate-master
----.github()
--------workflows()
----checkmate()
--------poet()
--------plot()
--------tf2()
--------core()
----tutorials()
--------tutorial_basic_tf2_example.ipynb(7KB)
----scripts()
--------poet_experiments()
--------test_tf2_execution.py(10KB)
--------poet_reference_mlp.py(1KB)
--------isolate_sqrtn_bug.py(2KB)
----LICENSE(11KB)
----setup.py(588B)
----README.md(4KB)
----Makefile(102B)
----tests()
--------test_cvxpy.py(884B)
--------test_linear.py(5KB)
--------test_tf2_extraction.py(573B)
----.gitignore(706B)