点燃:高级库可帮助您灵活透明地训练和评估PyTorch中的神经网络 源码

TL; DR Ignite是一个高级库,可帮助您灵活透明地训练和评估PyTorch中的神经网络。 点击图片查看完整的代码 产品特点 同时确保最大程度的控制和简化 库方法,无程序控制反转-在需要的位置和时间使用ignite 适用于指标,实验管理器和其他组件的可扩展API 目录 为什么点燃? Ignite是一个提供三个高级功能的库: 极其简单的引擎和事件系统 开箱即用的指标可轻松评估模型 内置处理程序以组成训练流水线,保存工件并记录参数和度量 简化的培训和验证循环 在历元和迭代中不再需要for/while循环的编码。 用户实例化引擎并运行它们。 例from ignite . engi