OneFlow v0.6.0 正式发布
今天是 OneFlow 开源的 528 天,OneFlow v0.6.0 正式发布。 点击“ 阅读原文 ”,欢迎下载体验最新版本。本次版本更新包括框架、模型和 OneFlow-ONNX 三大部分,主要有: 性能提升,包括静态图、动态图、算子性能、显存占用等方面 新增大量常用算子 完善静态图和 ConsistentTensor 功能 支持 OneFlow 作为 Nvidia Triton 的后端提供 Serving 功能 实现丰富的视觉预训练模型,与 torchvision、timm 对齐 实现更加完善的 OneFlow-ONNX 转换功能 以下为版本更新详情。 框架优化 1. 深度优化 nn.Graph 的性能 与 v0.5.0 相比, v0.6.0 的 nn.Graph 在 ResNet AMP 和 WDL 等模型上的训练速度提升了 10% 新版的动静转换功能的性能还有可优化的空间,近期着重优化了 nn.Graph 在高频率迭代训练场景下的性能 重新设计实现了 nn.Graph 的调度指令, 重构了 Actor Graph 与 Eager VM 的交互逻辑,使得 Graph 的 ru...

