pytorch中训练神经网络的直观库
ignis的Python项目详细描述
直观的库有助于在pytorch中训练神经网络
ignis
是一个高级库,可以帮助您编写紧凑但功能全面的训练循环,包括指标、早期停站,
以及深度学习库{a6}的模型检查点。
您可以根据自己的需要扩展ignis
。您可以通过扩展简单的
抽象类。
安装
- 安装pytorch。你可以在这里找到它:PyTorch
pip install ignis
示例
您可以在examples/
目录中找到示例
您还可以运行示例:python examples/iris.py
你可能想export PYTHONPATH=/path/to/this/directory
贡献
- 实现新功能
- 改进代码设计
- 改进评论和自述文件
- 测试