神经核
面向ML研究人员的轻量级PyTorch代码包装器。
NNCore是一个为机器学习和深度学习研究人员提供通用功能的库。 该项目旨在帮助用户在研究过程中更多地关注科学,而不是工程学。 基本功能包括但不限于:
通用I / O API
PyTorch中不包含的层和损失的有效实现
分布式培训的扩展方法
更强大的数据加载技术
该引擎可以接管整个培训和测试过程,并自动完成所有保姆工作(阶段控制,优化器配置,lr计划,检查点管理,度量和张量板编写等)。 有关详细信息,请参见。
请注意,该库中的某些方法可用于PyTorch 1.6+,但无需安装PyTorch。
持续集成
平台/ Python版本
3.6
3.7
3.8
3.9
Ubuntu 16.04
Ubuntu 18.04
的Ubuntu的20.04
macOS 10.15
视窗
安装
您可以直接从PyPI安装nncor
评论0
最新资源