Ignite是一个高级库,帮助你在PyTorch中训练神经网络
TL;DR Ignite 是一个高级库,可帮助在 PyTorch 中灵活透明地训练和评估神经网络。
单击图像查看完整代码 特点 比纯 PyTorch 更少的代码,同时确保最大程度的控制和简单性 库方法,没有程序的控制反转 - 随时随地使用 ignite 指标、实验管理器和其他组件的可扩展 API 目录为什么点燃?
安装入门 文档结构示例 教程 可复制的培训示例 交流 使用 Ignite 的贡献项目 关于团队 为什么选择 Ignite?
Ignite 是一个提供三个高级功能的库: 极其简单的引擎和事件系统 开箱即用的指标,可轻松评估模型 内置处理程序来组成训练管道、保存工件以及记录参数和指标 简化训练和验证循环不再编码 for/while 循环和迭代。
用户实例化引擎并运行它们。
Example from ignite.engine import Engine, Events, create_supervised_evaluator from ignite.metrics import Accuracy # 设置训
2021-09-07 15:57:56
10.12MB
机器学习
1