OneFlow v0.9.0更新,分布式深度学习框架
OneFlow v0.9.0更新,分布式深度学习框架 此更新包含 640 个提交和以下亮点: 1.OneFlow v0.9.0 增加了 86 个与 PyTorch 对齐的新 API 接口和运算符,并修复了 104 个与运算符兼容性相关的错误,从而提供了更好的 PyTorch API 和模型兼容性。在 v0.9.0 中,用户可以一键将更多 PyTorch 模型迁移到 OneFlow,获得更快的性能。 2.支持一键迁移Stable Diffusion、GLM、YOLOv5等到OneFlow。 3.更方便的模型迁移。Oneflow.load支持torch.save直接加载模型。 4.通过新增的oneflow.mock_torch模块和mock方法,oneflow可以在不改变原有PyTorch脚本的情况下,一键迁移包含多个脚本的复杂PyTorch模型。 5.Global Tensor增加了一系列方便分布式编程的接口和方法,并修复了已知的相关bug。 6.The Graph 发布了自动并行(version 1)的新特性,支持自动搜索指定 Placement 下最快的 SBP。使用 Global Tensor 编写分布式模型时,用户无需考虑并行性。 7.The Graph 添加了一系列与内存、执行速度、管道屏蔽和编译速度相关的优化,以提高性能并减少内存开销。 8.Graph提供了一系列辅助调试的功能,包括分析内存日志、显示编译阶段的进度、计算图等。 9.OneFlow IR 提供了更多的编译优化功能。 10.OneFlow的错误提示更加人性化,支持高亮显示错误内容,简化系统内部不必要的信息细节。就此而言,您可以直观地了解错误的位置和类型。 11.增加了一系列算子优化和系统优化,包括Eager指令调度、高性能CUDA内核、开放多个内存池等。 | #框架
在Telegram中查看相关推荐

🔍 发送关键词来寻找群组、频道或视频。
启动SOSO机器人