文件名称:KD_Lib:一个Pytorch知识蒸馏库,用于对知识蒸馏,修剪和量化领域中的工作进行基准测试和扩展
文件大小:364KB
文件格式:ZIP
更新时间:2024-05-24 04:39:29
benchmarking data-science machine-learning pytorch deep-learning-library
KD_Lib 一个PyTorch库可轻松促进自定义深度学习模型的知识提炼 安装 : 稳定释放 KD_Lib与Python 3.6或更高版本兼容,并且还取决于pytorch。 安装KD_Lib的最简单方法是使用pip(Python的首选软件包安装程序)。 $ pip install KD-Lib 请注意,KD_Lib是一个活动项目,并且定期发布新版本。 为了将KD_Lib升级到最新版本,请按以下方式使用pip。 $ pip install -U KD-Lib 从源代码构建 如果您打算安装该库的最新未发行版本(即从源代码),则只需执行以下操作: $ git clone https://github.com/SforAiDl/KD_Lib.git $ cd KD_Lib $ python setup.py install 用法 实施从最基本的知识版本,并绘制损失图 import to