定制适合自己的 Diffusers 扩散模型训练脚本

文章预览

↑ 点击 蓝字  关注极市平台 作者丨天才程序员周弈帆 来源丨天才程序员周弈帆 编辑丨极市平台 极市导读   本文介绍了如何定制Diffusers库的训练脚本，通过重构使其更兼容多种模型训练，分享了作者的GitHub代码示例。作者提出了一套设计原则，通过创建数据类和接口类来解耦训练逻辑，简化添加新任务的过程，并讨论了重构代码的体会。   >> 加入极市CV技术交流群，走在计算机视觉的最前沿 Diffusers 库为社区用户提供了多种扩散模型任务的训练脚本。每个脚本都平铺直叙，没有多余的封装，把训练的绝大多数细节都写在了一个脚本里。这种设计既能让入门用户在不阅读源码的前提下直接用脚本训练，又方便高级用户直接修改脚本。 可是，这种设计就是最好的吗？关于训练脚本的最佳设计风格，社区用户们往往各执一词。有人更喜欢更贴近 PyTorch ………………………………