只是另一个dl库。
kekas的Python项目详细描述
凯卡斯
别紧张。
kekas是在pytorch上训练神经网络的一个简单工具。
我尽量把事情简单化。
安装
pip install kekas
混合精度要求
kekas使用https://github.com/NVIDIA/apex库进行混合精度训练,因此请遵循其repo中的安装说明。
功能
- 好名字
- 混合精度(fp16)
- 学习率查找器
- 单周期政策
- 保存最佳检查点
- 提前停车
- TTA
- 冻结/解冻
- 轻松定制
快速入门指南
我不喜欢快速入门指南,我认为它们会产生更多的问题 而不是答案。
相反,我创建了一个详细的Tutorial notebook。读一下。
贡献指南
贡献一些好的东西,不要贡献任何坏的东西。
待办事项
- []ADAMW
- [X]FP16支持
- []区别学习率
- []文档:kekeke:
引用
如果您发现此库对您的研究有用,请考虑引用:
@misc{aleksandr belskikh_2019,
author = {Aleksandr Belskikh},
title = {{kekas: Just another DL library}},
month = jul,
year = 2019,
doi = {10.5281/zenodo.2577861},
version = {0.1.18},
publisher = {Zenodo},
url = {https://doi.org/10.5281/zenodo.2577861},
}